diff --git a/packages/autorest.python/test/unittests/test_m2r.py b/packages/autorest.python/test/unittests/test_m2r.py
index 8ac1287dda..f61941a3b7 100644
--- a/packages/autorest.python/test/unittests/test_m2r.py
+++ b/packages/autorest.python/test/unittests/test_m2r.py
@@ -3,7 +3,7 @@
 # Licensed under the MIT License. See License.txt in the project root for
 # license information.
 # --------------------------------------------------------------------------
-from autorest.m2r import M2R
+from pygen.m2r import M2R
 
 
 _MD_LINK = "[inline link](https://github.com/Azure/autorest.python)"
diff --git a/packages/autorest.python/test/unittests/test_name_converter.py b/packages/autorest.python/test/unittests/test_name_converter.py
index 1e650033e9..930eb0fc60 100644
--- a/packages/autorest.python/test/unittests/test_name_converter.py
+++ b/packages/autorest.python/test/unittests/test_name_converter.py
@@ -3,8 +3,8 @@
 # Licensed under the MIT License. See License.txt in the project root for
 # license information.
 # --------------------------------------------------------------------------
-from autorest.preprocess import PreProcessPlugin
-from autorest.preprocess.python_mappings import PadType
+from pygen.preprocess import PreProcessPlugin
+from pygen.preprocess.python_mappings import PadType
 
 
 def pad_reserved_words(name: str, pad_type: PadType) -> str:
diff --git a/packages/autorest.python/test/unittests/test_optional_return_type.py b/packages/autorest.python/test/unittests/test_optional_return_type.py
index 1a4cfae46e..26f413714c 100644
--- a/packages/autorest.python/test/unittests/test_optional_return_type.py
+++ b/packages/autorest.python/test/unittests/test_optional_return_type.py
@@ -5,7 +5,7 @@
 # --------------------------------------------------------------------------
 
 import pytest
-from autorest.codegen.models import (
+from pygen.codegen.models import (
     Operation,
     LROOperation,
     PagingOperation,
@@ -15,8 +15,8 @@
     RequestBuilder,
     Client,
 )
-from autorest.codegen.models.parameter_list import RequestBuilderParameterList
-from autorest.codegen.models.primitive_types import StringType
+from pygen.codegen.models.parameter_list import RequestBuilderParameterList
+from pygen.codegen.models.primitive_types import StringType
 
 
 @pytest.fixture
diff --git a/packages/autorest.python/test/unittests/test_parameter_ordering.py b/packages/autorest.python/test/unittests/test_parameter_ordering.py
index 8c68fd1486..4f64c991ad 100644
--- a/packages/autorest.python/test/unittests/test_parameter_ordering.py
+++ b/packages/autorest.python/test/unittests/test_parameter_ordering.py
@@ -3,8 +3,8 @@
 # Licensed under the MIT License. See License.txt in the project root for
 # license information.
 # --------------------------------------------------------------------------
-from autorest.codegen.models import Parameter, AnyType, CodeModel, StringType
-from autorest.codegen.models.parameter_list import ParameterList
+from pygen.codegen.models import Parameter, AnyType, CodeModel, StringType
+from pygen.codegen.models.parameter_list import ParameterList
 
 
 def get_code_model():
diff --git a/packages/autorest.python/test/unittests/test_sort_schema.py b/packages/autorest.python/test/unittests/test_sort_schema.py
index 156b162598..253b83f100 100644
--- a/packages/autorest.python/test/unittests/test_sort_schema.py
+++ b/packages/autorest.python/test/unittests/test_sort_schema.py
@@ -4,7 +4,7 @@
 # license information.
 # --------------------------------------------------------------------------
 
-from autorest.codegen.models import CodeModel, DPGModelType
+from pygen.codegen.models import CodeModel, DPGModelType
 
 
 def get_code_model():