From c51c5a2b3447f61f0694c298e9d6f4220d0d70c6 Mon Sep 17 00:00:00 2001
From: Bertie690 <taylormw163@gmail.com>
Date: Sun, 3 Aug 2025 23:13:25 -0400
Subject: [PATCH] hopefullt fixed tests

---
 test/abilities/arena-trap.test.ts | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/test/abilities/arena-trap.test.ts b/test/abilities/arena-trap.test.ts
index 2c0ff37eaed..4f6a4bb5125 100644
--- a/test/abilities/arena-trap.test.ts
+++ b/test/abilities/arena-trap.test.ts
@@ -29,8 +29,8 @@ describe("Abilities - Arena Trap", () => {
     game = new GameManager(phaserGame);
     game.override
       .ability(AbilityId.ARENA_TRAP)
+      .enemyAbility(AbilityId.ARENA_TRAP)
       .enemySpecies(SpeciesId.RALTS)
-      .enemyAbility(AbilityId.BALL_FETCH)
       .enemyMoveset(MoveId.SPLASH);
   });
 
@@ -63,7 +63,7 @@ describe("Abilities - Arena Trap", () => {
   });
 
   it("should interrupt player switch attempt and display message", async () => {
-    game.override.battleStyle("single").enemyAbility(AbilityId.ARENA_TRAP);
+    game.override.battleStyle("single");
     await game.classicMode.startBattle([SpeciesId.DUGTRIO, SpeciesId.GOTHITELLE]);
 
     const enemy = game.field.getEnemyPokemon();
@@ -74,6 +74,7 @@ describe("Abilities - Arena Trap", () => {
       expect(game.scene.currentBattle.turnCommands[0]).toBeNull();
 
       // back out and cancel the switch to avoid timeout
+      (game.scene.ui.getHandler() as CommandUiHandler).processInput(Button.ACTION);
       (game.scene.ui.getHandler() as CommandUiHandler).processInput(Button.CANCEL);
       game.move.use(MoveId.SPLASH);
     });