diff --git a/ax/preview/api/configs.py b/ax/preview/api/configs.py
index 43c8ec2f850..559e9af1730 100644
--- a/ax/preview/api/configs.py
+++ b/ax/preview/api/configs.py
@@ -10,7 +10,7 @@
 from typing import Any, Callable, List, Mapping, Sequence
 
 from ax.preview.api.types import TParameterValue
-from ax.storage.registry_bundle import RegistryBundle
+from ax.storage.registry_bundle import RegistryBundleBase
 
 
 class ParameterType(Enum):
@@ -163,4 +163,4 @@ class OrchestrationConfig:
 class StorageConfig:
     creator: Callable[..., Any] | None = None  # pyre-fixme[4]
     url: str | None = None
-    registry_bundle: RegistryBundle | None = None
+    registry_bundle: RegistryBundleBase | None = None