diff --git a/plugin/napari_lattice/shape_selector.py b/plugin/napari_lattice/shape_selector.py
index 4acaf81..d51e7fc 100644
--- a/plugin/napari_lattice/shape_selector.py
+++ b/plugin/napari_lattice/shape_selector.py
@@ -8,7 +8,7 @@
 from collections import defaultdict
 from contextlib import contextmanager
 
-from plugin.napari_lattice.utils import get_viewer
+from napari_lattice.utils import get_viewer
 
 if TYPE_CHECKING:
     from napari.utils.events.event import Event