diff --git a/sdk/storage/azure-storage-blob/tests/test_blob_storage_account.py b/sdk/storage/azure-storage-blob/tests/test_blob_storage_account.py
index 1f93d40bcdf9..b062ed2e9b12 100644
--- a/sdk/storage/azure-storage-blob/tests/test_blob_storage_account.py
+++ b/sdk/storage/azure-storage-blob/tests/test_blob_storage_account.py
@@ -4,7 +4,7 @@
 # license information.
 # --------------------------------------------------------------------------
 from azure.storage.blob import BlobServiceClient, StandardBlobTier
-from azure.storage.blob._generated.models import RehydratePriority
+from azure.storage.blob._generated.azure.storage.blobs.models import RehydratePriority
 
 from devtools_testutils import recorded_by_proxy
 from devtools_testutils.storage import StorageRecordedTestCase
diff --git a/sdk/storage/azure-storage-blob/tests/test_blob_storage_account_async.py b/sdk/storage/azure-storage-blob/tests/test_blob_storage_account_async.py
index bb5638227afd..43cf942b7c66 100644
--- a/sdk/storage/azure-storage-blob/tests/test_blob_storage_account_async.py
+++ b/sdk/storage/azure-storage-blob/tests/test_blob_storage_account_async.py
@@ -6,7 +6,7 @@
 from azure.core.pipeline.transport import AioHttpTransport
 from azure.storage.blob import StandardBlobTier
 from azure.storage.blob.aio import BlobServiceClient
-from azure.storage.blob._generated.models import RehydratePriority
+from azure.storage.blob._generated.azure.storage.blobs.models import RehydratePriority
 
 from devtools_testutils.aio import recorded_by_proxy_async
 from devtools_testutils.storage.aio import AsyncStorageRecordedTestCase
diff --git a/sdk/storage/azure-storage-blob/tests/test_common_blob.py b/sdk/storage/azure-storage-blob/tests/test_common_blob.py
index 17ea27de2141..ad179e7f26cf 100644
--- a/sdk/storage/azure-storage-blob/tests/test_common_blob.py
+++ b/sdk/storage/azure-storage-blob/tests/test_common_blob.py
@@ -49,7 +49,7 @@
     generate_blob_sas,
     generate_container_sas,
     upload_blob_to_url)
-from azure.storage.blob._generated.models import RehydratePriority
+from azure.storage.blob._generated.azure.storage.blobs.models import RehydratePriority
 
 from devtools_testutils import FakeTokenCredential, recorded_by_proxy
 from devtools_testutils.storage import StorageRecordedTestCase