diff --git a/sdk/eventhub/azure-eventhub-checkpointstoreblob-aio/azure/eventhub/extensions/__init__.py b/sdk/eventhub/azure-eventhub-checkpointstoreblob-aio/azure/eventhub/extensions/__init__.py
index 34913fb394d7a..f70db896c6150 100644
--- a/sdk/eventhub/azure-eventhub-checkpointstoreblob-aio/azure/eventhub/extensions/__init__.py
+++ b/sdk/eventhub/azure-eventhub-checkpointstoreblob-aio/azure/eventhub/extensions/__init__.py
@@ -2,3 +2,4 @@
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # Licensed under the MIT License. See License.txt in the project root for license information.
 # --------------------------------------------------------------------------------------------
+__path__ = __import__("pkgutil").extend_path(__path__, __name__)  # type: ignore
diff --git a/sdk/eventhub/azure-eventhub-checkpointstoreblob/azure/eventhub/extensions/__init__.py b/sdk/eventhub/azure-eventhub-checkpointstoreblob/azure/eventhub/extensions/__init__.py
index e69de29bb2d1d..f70db896c6150 100644
--- a/sdk/eventhub/azure-eventhub-checkpointstoreblob/azure/eventhub/extensions/__init__.py
+++ b/sdk/eventhub/azure-eventhub-checkpointstoreblob/azure/eventhub/extensions/__init__.py
@@ -0,0 +1,5 @@
+# --------------------------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# --------------------------------------------------------------------------------------------
+__path__ = __import__("pkgutil").extend_path(__path__, __name__)  # type: ignore
diff --git a/sdk/eventhub/azure-eventhub/azure/eventhub/__init__.py b/sdk/eventhub/azure-eventhub/azure/eventhub/__init__.py
index 5ab6c4f5ef5d1..b97277f9335b0 100644
--- a/sdk/eventhub/azure-eventhub/azure/eventhub/__init__.py
+++ b/sdk/eventhub/azure-eventhub/azure/eventhub/__init__.py
@@ -2,9 +2,6 @@
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # Licensed under the MIT License. See License.txt in the project root for license information.
 # --------------------------------------------------------------------------------------------
-
-__path__ = __import__("pkgutil").extend_path(__path__, __name__)  # type: ignore
-
 from uamqp import constants
 from ._common import EventData, EventDataBatch
 from ._version import VERSION