diff --git a/rpi/antRender.py b/rpi/antRender.py
index 4d17704..b233850 100644
--- a/rpi/antRender.py
+++ b/rpi/antRender.py
@@ -1,6 +1,6 @@
 from rgbmatrix import RGBMatrix, RGBMatrixOptions
-from Point2D import Point2D
-from ProotState import ProotState
+import Point2D
+import ProotState
 
 from PIL import Image
 import paho.mqtt.client as mqtt
@@ -99,7 +99,7 @@ for alpha in range(0,11):
 endT = curr_time = round(time.time()*1000)
 print("populating matrices for each blink frame took: " + str(endT - startT) + " ms")
 
-proot_state = ProotState()
+proot_state = ProotState.ProotState()
 proot_state.set_blinks_frames_ready(True)
 proot_state.blink()
 
@@ -114,7 +114,7 @@ def on_connect(client, userdata, flags, response_code):
 def on_message(client, userdata, message):
     print("Received message '" + str(message.payload) + "' on topic '"
           + message.topic + "' with QoS " + str(message.qos))
-    proot_state = ProotState()
+    proot_state = ProotState.ProotState()
     proot_state.blink()
 
 # MQTT broker configuration