diff --git a/libs/computer-server/computer_server/diorama/diorama.py b/libs/computer-server/computer_server/diorama/diorama.py
index b0564d4a..987be19c 100644
--- a/libs/computer-server/computer_server/diorama/diorama.py
+++ b/libs/computer-server/computer_server/diorama/diorama.py
@@ -9,7 +9,7 @@ import io
 from typing import Union
 from PIL import Image, ImageDraw
 
-from draw import capture_all_apps, AppActivationContext, get_frontmost_and_active_app, get_all_windows, get_running_apps
+from computer_server.diorama.draw import capture_all_apps, AppActivationContext, get_frontmost_and_active_app, get_all_windows, get_running_apps
 
 from diorama_computer import DioramaComputer
 from computer_server.handlers.macos import *
diff --git a/libs/computer-server/computer_server/diorama/macos.py b/libs/computer-server/computer_server/diorama/macos.py
index 3d5a6f17..f5e77178 100644
--- a/libs/computer-server/computer_server/diorama/macos.py
+++ b/libs/computer-server/computer_server/diorama/macos.py
@@ -2,8 +2,8 @@ import platform
 import sys
 import platform
 import inspect
-from .diorama import Diorama
-from .base import BaseDioramaHandler
+from computer_server.diorama.diorama import Diorama
+from computer_server.diorama.base import BaseDioramaHandler
 
 class MacOSDioramaHandler(BaseDioramaHandler):
     """Handler for Diorama commands on macOS, using local diorama module."""