diff --git a/phoenix/widgets/__init__.py b/phoenix/widgets/__init__.py
index 7284632b8d..f07b420321 100644
--- a/phoenix/widgets/__init__.py
+++ b/phoenix/widgets/__init__.py
@@ -1,3 +1 @@
-from .umapWidget import demo_json
-from .umapWidget import UMAPWidget
-
+from .umapWidget import UMAPWidget, demo_json