diff --git a/spacy/tokens/token.pyx b/spacy/tokens/token.pyx index 6c1e1b06f..be972f222 100644 --- a/spacy/tokens/token.pyx +++ b/spacy/tokens/token.pyx @@ -35,11 +35,11 @@ cdef class Token: @classmethod def get_extension(cls, name): - return Underscore.span_extensions.get(name) + return Underscore.token_extensions.get(name) @classmethod def has_extension(cls, name): - return name in Underscore.span_extensions + return name in Underscore.token_extensions @classmethod def remove_extension(cls, name):