diff --git a/rpi/ProotState.py b/rpi/ProotState.py
index a18e5cf..91c64a3 100644
--- a/rpi/ProotState.py
+++ b/rpi/ProotState.py
@@ -1,5 +1,6 @@
 import time
 import random
+from PIL import Image
 
 class ProotState:
     _instance = None