diff --git a/overcooked_simulator/overcooked_environment.py b/overcooked_simulator/overcooked_environment.py index 00a5794c3502ef44aa54663b245d922bb2d16164..847db60b01065440de5d5e71f36534dcfccd4bcc 100644 --- a/overcooked_simulator/overcooked_environment.py +++ b/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: