diff --git a/src/gamestates/battle.py b/src/gamestates/battle.py
index b3f6dd3..fc52b28 100644
--- a/src/gamestates/battle.py
+++ b/src/gamestates/battle.py
@@ -285,13 +285,13 @@ def update(game, dt):
     for m in menu:
         m.update()
 
-    if len(enemies) != 0 and all(e.ko for e in enemies):
+    if len(enemies) != 0 and all(e.ko or e.stone for e in enemies):
         print("\nVICTORY! all enemies were defeated.")
         battle_music.stop()
         gamestate.switch_gamestate(game,"overworld")
         return
 
-    if len(party_members) != 0 and all(a.ko for a in party_members):
+    if len(party_members) != 0 and all(a.ko or a.stone for a in party_members):
         print("\nGAME OVER! the party was defeated.")
         battle_music.stop()
         gamestate.switch_gamestate(game,"gameover")