diff --git a/cooperative_cuisine/pygame_2d_vis/drawing.py b/cooperative_cuisine/pygame_2d_vis/drawing.py
index 84176c29539109b062cc9231a03b014e1eb505d3..88ece8dafb45c3a33476dd7f727e66017a2426b6 100644
--- a/cooperative_cuisine/pygame_2d_vis/drawing.py
+++ b/cooperative_cuisine/pygame_2d_vis/drawing.py
@@ -943,9 +943,7 @@ class Visualizer:
 
     def get_state_image(self, state: dict,
                         cache_flags: CacheFlags = CacheFlags.COUNTERS | CacheFlags.BACKGROUND) -> npt.NDArray[np.uint8]:
-        screen = self.draw_gamescreen(
-            state, [0 for _ in state["players"]], cache_flags=cache_flags
-        )
+        screen = self.draw_gamescreen(state, [0 for _ in state["players"]], cache_flags=cache_flags)
         return pygame.surfarray.pixels3d(screen)
 
     def draw_recipe_image(
diff --git a/cooperative_cuisine/pygame_2d_vis/drawing_perf_benchmark.py b/cooperative_cuisine/pygame_2d_vis/drawing_perf_benchmark.py
index 42cf742354e798f91111e5c6cbbd2cf76e41a3d6..d12d3c67239dacf482f4dae6c75964a1b8c2c2ef 100644
--- a/cooperative_cuisine/pygame_2d_vis/drawing_perf_benchmark.py
+++ b/cooperative_cuisine/pygame_2d_vis/drawing_perf_benchmark.py
@@ -47,11 +47,10 @@ def run_benchmark(GRID_SIZE, ITERS, OPTIMIZATIONS):
 
         time_start = time.perf_counter()
         observation = visualizer.get_state_image(state=json_dict, cache_flags=OPTIMIZATIONS["cache_counters"])
-        img = observation.astype(np.uint8).transpose((1, 0, 2))
 
-        img = cv2.resize(img, (400, 400), interpolation=cv2.INTER_NEAREST)
-        cv2.imshow("env", img[:, :, ::-1])
-        cv2.waitKey(1)
+        # img = cv2.resize(img, (400, 400), interpolation=cv2.INTER_NEAREST)
+        # cv2.imshow("env", img[:, :, ::-1])
+        # cv2.waitKey(1)
         durations.append(time.perf_counter() - time_start)
 
     print(f"Mean duration: %.6fs" % np.mean(durations))
@@ -75,7 +74,7 @@ if __name__ == "__main__":
     with open(ROOT_DIR / "pygame_2d_vis" / "visualization.yaml", "r") as file:
         visualization_config = yaml.safe_load(file)
 
-    GRID_SIZE = 100
+    GRID_SIZE = 64
     ITERS = 1000
 
     OPTIMIZATIONS = {"use_array3d": True, "cache_counters": CacheFlags.BACKGROUND | CacheFlags.COUNTERS}