From 580702c9c8f41bc69eafc7a7bfa8a07b47fd95d5 Mon Sep 17 00:00:00 2001
From: NightKev <34855794+DayKev@users.noreply.github.com>
Date: Mon, 18 Aug 2025 18:48:32 -0700
Subject: [PATCH] Update test

---
 test/abilities/embody-aspect.test.ts | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/test/abilities/embody-aspect.test.ts b/test/abilities/embody-aspect.test.ts
index 00fe53071fb..43e4f9015dd 100644
--- a/test/abilities/embody-aspect.test.ts
+++ b/test/abilities/embody-aspect.test.ts
@@ -2,7 +2,6 @@ import { AbilityId } from "#enums/ability-id";
 import { MoveId } from "#enums/move-id";
 import { SpeciesId } from "#enums/species-id";
 import { Stat } from "#enums/stat";
-import { QuietFormChangePhase } from "#phases/quiet-form-change-phase";
 import { GameManager } from "#test/test-utils/game-manager";
 import Phaser from "phaser";
 import { afterEach, beforeAll, beforeEach, describe, expect, it } from "vitest";
@@ -27,7 +26,7 @@ describe("Ability - Embody Aspect", () => {
   beforeEach(() => {
     game = new GameManager(phaserGame);
     game.override
-      .moveset([MoveId.SPLASH])
+      .moveset(MoveId.SPLASH)
       .ability(AbilityId.EMBODY_ASPECT_TEAL)
       .battleStyle("single")
       .criticalHits(false)
@@ -38,16 +37,17 @@ describe("Ability - Embody Aspect", () => {
   it("Embody Aspect should activate on switching in if user is terrestrialized", async () => {
     await game.classicMode.startBattle([SpeciesId.OGERPON, SpeciesId.ABOMASNOW]);
 
-    const Ogerpon = game.scene.getPlayerParty()[0];
-    expect(Ogerpon.formIndex).toBe(baseForm);
+    const ogerpon = game.field.getPlayerPokemon();
+    expect(ogerpon.formIndex).toBe(baseForm);
+    expect(ogerpon.getStatStage(Stat.SPD)).toBe(0);
 
     //Also terastallizes Ogerpon
     game.move.selectWithTera(MoveId.SPLASH);
 
-    await game.phaseInterceptor.to(QuietFormChangePhase);
-    expect(Ogerpon.formIndex).toBe(teraForm);
+    await game.phaseInterceptor.to("QuietFormChangePhase");
+    expect(ogerpon.formIndex).toBe(teraForm);
     await game.toNextTurn();
-    expect(Ogerpon.getStatStage(Stat.SPD)).toBe(1);
+    expect(ogerpon.getStatStage(Stat.SPD)).toBe(1);
 
     //Switch Ogerpon out
     game.doSwitchPokemon(1);
@@ -58,6 +58,6 @@ describe("Ability - Embody Aspect", () => {
     await game.toNextTurn();
 
     //Ability activated
-    expect(Ogerpon.getStatStage(Stat.SPD)).toBe(1);
+    expect(ogerpon.getStatStage(Stat.SPD)).toBe(1);
   });
 });