diff --git a/olive/passes/__init__.py b/olive/passes/__init__.py
index 06e5bfc31b..ff11be1b28 100644
--- a/olive/passes/__init__.py
+++ b/olive/passes/__init__.py
@@ -12,4 +12,5 @@
     "PassParamDefault",
     "PassModuleConfig",
     "FullPassConfig",
+    "REGISTRY",
 ]
diff --git a/olive/platform_sdk/qualcomm/utils/__init__.py b/olive/platform_sdk/qualcomm/utils/__init__.py
index 22ca05454a..862c45ce31 100644
--- a/olive/platform_sdk/qualcomm/utils/__init__.py
+++ b/olive/platform_sdk/qualcomm/utils/__init__.py
@@ -2,6 +2,3 @@
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # Licensed under the MIT License.
 # --------------------------------------------------------------------------
-
-from olive.platform_sdk.qualcomm.constants import *  # noqa: F403
-from olive.platform_sdk.qualcomm.env import *  # noqa: F403
diff --git a/olive/strategy/search_algorithm/__init__.py b/olive/strategy/search_algorithm/__init__.py
index e5843881b9..54e7f05ff8 100644
--- a/olive/strategy/search_algorithm/__init__.py
+++ b/olive/strategy/search_algorithm/__init__.py
@@ -9,4 +9,4 @@
 
 REGISTRY = SearchAlgorithm.registry
 
-__all__ = ["SearchAlgorithm", "ExhaustiveSearchAlgorithm", "RandomSearchAlgorithm", "TPESearchAlgorithm"]
+__all__ = ["SearchAlgorithm", "ExhaustiveSearchAlgorithm", "RandomSearchAlgorithm", "TPESearchAlgorithm", "REGISTRY"]