Update overcooked_environment.py

overcooked_simulator/overcooked_environment.py

@@ -904,9 +904,9 @@ class Environment:
             name = "Meat"
 
         if name in self.vector_state_generation.meals:
-            idx = self.vector_state_generation.meals.index(name)
+            idx = 3 + self.vector_state_generation.meals.index(name)
         elif name in self.vector_state_generation.ingredients:
-            idx = len(
+            idx = 3 + len(
                 self.vector_state_generation.meals
             ) + self.vector_state_generation.ingredients.index(name)
         else: