diff --git a/spacy/tests/pipeline/test_textcat.py b/spacy/tests/pipeline/test_textcat.py
index d352d8fe2..8a0c1a976 100644
--- a/spacy/tests/pipeline/test_textcat.py
+++ b/spacy/tests/pipeline/test_textcat.py
@@ -44,7 +44,7 @@ TRAIN_DATA_MULTI_LABEL = [
 
 lazy_init_model_config = """
 [model]
-@architectures = "LazyInitTok2Vec.v1"
+@architectures = "test.LazyInitTok2Vec.v1"
 width = 96
 """
 LAZY_INIT_TOK2VEC_MODEL = Config().from_str(lazy_init_model_config)["model"]
diff --git a/spacy/tests/tok2vec.py b/spacy/tests/tok2vec.py
index 5e1a29e76..7e7b689eb 100644
--- a/spacy/tests/tok2vec.py
+++ b/spacy/tests/tok2vec.py
@@ -7,7 +7,7 @@ from spacy.tokens import Doc
 from spacy.util import registry
 
 
-@registry.architectures("LazyInitTok2Vec.v1")
+@registry.architectures("test.LazyInitTok2Vec.v1")
 def build_lazy_init_tok2vec(*, width: int) -> Model[List[Doc], List[Floats2d]]:
     """tok2vec model of which the output size is only known after
     initialization. This implementation does not output meaningful