diff --git a/openflexure_microscope/stage/__init__.py b/openflexure_microscope/stage/__init__.py
index 5aab8d20..9ef9d9f3 100644
--- a/openflexure_microscope/stage/__init__.py
+++ b/openflexure_microscope/stage/__init__.py
@@ -1 +1 @@
-from . import openflexure
\ No newline at end of file
+from . import base, mock, sanga
\ No newline at end of file