diff --git a/libs/computer-server/computer_server/diorama/__init__.py b/libs/computer-server/computer_server/diorama/__init__.py
index 3608c381..e69de29b 100644
--- a/libs/computer-server/computer_server/diorama/__init__.py
+++ b/libs/computer-server/computer_server/diorama/__init__.py
@@ -1,3 +0,0 @@
-from .diorama import Diorama
-
-__all__ = ["Diorama"]
diff --git a/libs/computer-server/computer_server/diorama/draw.py b/libs/computer-server/computer_server/diorama/draw.py
index d4097164..15d00fcc 100644
--- a/libs/computer-server/computer_server/diorama/draw.py
+++ b/libs/computer-server/computer_server/diorama/draw.py
@@ -36,8 +36,7 @@ logging.basicConfig(
 )
 logger = logging.getLogger("diorama.draw")
 
-
-from safezone import (
+from computer_server.diorama.safezone import (
     get_menubar_bounds,
     get_dock_bounds,
 )