diff --git a/BattlePongAI/main.py b/BattlePongAI/main.py
index 25bb3fd9e163d899d8b1cfe9d510928dee0e9d25..9b5f2856c63f74457273727287fb81fe1edc1c99 100644
--- a/BattlePongAI/main.py
+++ b/BattlePongAI/main.py
@@ -15,7 +15,8 @@ async def main():
     recieved_values = await gym.reset()
     data = recieved_values[0]
     print('main data', data)
-    image_mode = 'RGB'
+    #image_mode = 'RGB'
+    image_mode = 'L'
     try:
         # Test show Image
         si.show_image(image_mode, data['info']['screenshot']['width'], data['info']['screenshot']['height'],