diff --git a/main.py b/main.py
index aa74fe3..dec70f7 100644
--- a/main.py
+++ b/main.py
@@ -52,7 +52,7 @@ def get_track():
     #image.save("thumbnail.jpg","JPEG")
     print('set image')
     image = Image.new('RGB', (32, 32), color = 'red')
-    screen.SetImage(image.convert('RGB'))
+    self.screen.SetImage(image.convert('RGB'))
     print('image set')
 
 #if token: