diff --git a/code/main.gd b/code/main.gd
index 8d0814e..dc8dff7 100644
--- a/code/main.gd
+++ b/code/main.gd
@@ -42,10 +42,11 @@ func _process(delta):
 func _on_player_death():
     var i = 0
     for player in players.get_children():
+        player.ai_controller.done = true
+        player.ai_controller.needs_reset = true
+        player.ai_controller.reset()
         player.position = player_starting_pos[i]
         player.health = 100
-        player.ai_controller.done = true
-        player.ai_controller.reset()
         i += 1
     var j = 0
     for bamboo in bamboos.get_children():
diff --git a/logs/sb3/policy_big_tanh_1/events.out.tfevents.1716360819.valanixos.270988.0 b/logs/sb3/policy_big_tanh_1/events.out.tfevents.1716360819.valanixos.270988.0
index 38de345..6fcb42f 100644
Binary files a/logs/sb3/policy_big_tanh_1/events.out.tfevents.1716360819.valanixos.270988.0 and b/logs/sb3/policy_big_tanh_1/events.out.tfevents.1716360819.valanixos.270988.0 differ
diff --git a/logs/sb3/policy_big_tanh_2/events.out.tfevents.1716640056.valanixos.6476.0 b/logs/sb3/policy_big_tanh_2/events.out.tfevents.1716640056.valanixos.6476.0
new file mode 100644
index 0000000..264474f
Binary files /dev/null and b/logs/sb3/policy_big_tanh_2/events.out.tfevents.1716640056.valanixos.6476.0 differ