diff --git a/rpi/antRender.py b/rpi/antRender.py
index 924cbdf..f3cea24 100644
--- a/rpi/antRender.py
+++ b/rpi/antRender.py
@@ -1,6 +1,6 @@
 from rgbmatrix import RGBMatrix, RGBMatrixOptions
 import Point2D
-import ProotState
+from ProotState import ProotState
 
 import time
 import random
@@ -120,7 +120,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.ProotState()
+proot_state = ProotState()
 proot_state.set_matrix(matrix)
 proot_state.set_blinks_frames_ready(True)
 proot_state.blink()
@@ -136,7 +136,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.ProotState()
+    proot_state = ProotState()
     proot_state.blink()
 
 # MQTT broker configuration