diff --git a/test/codegen/codegen.js b/test/codegen/codegen.js
index de74a9d3..4101d760 100644
--- a/test/codegen/codegen.js
+++ b/test/codegen/codegen.js
@@ -44,7 +44,7 @@ describe('codegen', function () {
         versionedModelManager.addCTOModel(cto, 'hr.cto');
 
         const unversionedBaseCto = base_cto.replace('namespace org.acme.hr.base@1.0.0', 'namespace org.acme.hr.base');
-        const unversionedCto = cto.replace('namespace org.acme.hr@1.0.0', 'namespace org.acme.hr');
+const unversionedCto = cto.replace('namespace org.acme.hr@1.0.0', 'namespace org.acme.hr').replace('import org.acme.hr.base@1.0.0', 'import org.acme.hr.base');
         unversionedModelManager.addCTOModel(unversionedBaseCto, 'hr_base.cto');
         unversionedModelManager.addCTOModel(unversionedCto, 'hr.cto');
     });