diff --git a/nwb_linkml/src/nwb_linkml/generators/pydantic.py b/nwb_linkml/src/nwb_linkml/generators/pydantic.py
index ce3f1e2..6732063 100644
--- a/nwb_linkml/src/nwb_linkml/generators/pydantic.py
+++ b/nwb_linkml/src/nwb_linkml/generators/pydantic.py
@@ -266,7 +266,7 @@ class NWBPydanticGenerator(PydanticGenerator):
                 version = version_module_case(self.versions[schema_name])
                 if self.split:
                     local_mod_name = '...' + module_case(schema_name) + '.' + version + '.' + module_case(module_name)
-                else:
+                else: # pragma: no cover
                     local_mod_name = '...' + module_case(schema_name) + '.' + version + '.' + 'namespace'
 
             else:
diff --git a/nwb_linkml/tests/test_generators/test_generator_pydantic.py b/nwb_linkml/tests/test_generators/test_generator_pydantic.py
index 507dcd4..ae2cb51 100644
--- a/nwb_linkml/tests/test_generators/test_generator_pydantic.py
+++ b/nwb_linkml/tests/test_generators/test_generator_pydantic.py
@@ -165,7 +165,7 @@ def test_versions(linkml_schema):
     core_str = NWBPydanticGenerator(
         str(linkml_schema.core_path),
         versions={'imported': 'v4.2.0'},
-        split=False
+        split=True
     ).serialize()
 
     # the import should be like
@@ -175,19 +175,6 @@ def test_versions(linkml_schema):
     match = re.findall(r'from \.\.\.imported\.v4_2_0.*?MainThing.*?\)', core_str, flags=re.DOTALL)
     assert len(match) == 1
 
-    core_str = NWBPydanticGenerator(
-        str(linkml_schema.core_path),
-        versions={'imported': 'v4.2.0'},
-        split=True
-    ).serialize()
-
-    # the import should be like
-    # from ...imported.v4_2_0.namespace import (
-    #     MainThing
-    # )
-    match = re.findall(r'from \.\.\.imported\.v4_2_0\.namespace.*?MainThing.*?\)', core_str, flags=re.DOTALL)
-    assert len(match) == 1
-
 
 def test_arraylike(imported_schema):
     """