diff --git a/geary-tests/src/test/kotlin/com/mineinabyss/geary/papermc/tracking/entities/EntityTrackingTests.kt b/geary-tests/src/test/kotlin/com/mineinabyss/geary/papermc/tracking/entities/EntityTrackingTests.kt
index 4ce1361..28b4e22 100644
--- a/geary-tests/src/test/kotlin/com/mineinabyss/geary/papermc/tracking/entities/EntityTrackingTests.kt
+++ b/geary-tests/src/test/kotlin/com/mineinabyss/geary/papermc/tracking/entities/EntityTrackingTests.kt
@@ -65,7 +65,7 @@ class EntityTrackingTests: MockedServerTest() {
         val pig = world.spawn(world.spawnLocation, Pig::class.java)
 
         // act
-        EntityAddToWorldEvent(pig).callEvent() // Not called by MockBukkit
+        EntityAddToWorldEvent(pig, world).callEvent() // Not called by MockBukkit
 
         // assert
         val gearyPig = pig.toGearyOrNull().shouldNotBeNull()
@@ -105,7 +105,7 @@ class EntityTrackingTests: MockedServerTest() {
         @Test
         fun `should persist components on entities when chunk unloaded`() {
             val pig = world.spawn(world.spawnLocation, Pig::class.java)
-            EntityAddToWorldEvent(pig).callEvent()
+            EntityAddToWorldEvent(pig, world).callEvent()
 
             testPersistence(
                 pig.toGeary(),