diff --git a/starvla_inference_server.py b/starvla_inference_server.py
index bb07bd3..3f8c2cd 100644
--- a/starvla_inference_server.py
+++ b/starvla_inference_server.py
@@ -129,7 +129,7 @@ class StarvlaInferenceServer:
     def inference(self, observation: dict) -> dict:
         
         img_head, state_vec, prompt = \
-            self.parse_observation(observation)
+            self.parse_observation(observation, target_size=(410, 224))
         vla_input = {
             # "batch_images": [[img_left, img_right, img_wrist]],
             "image": [img_head],