From 14ea5f1c72a2faedba0caaa7d54c5947bd0b881c Mon Sep 17 00:00:00 2001
From: WinDanesz <31292708+WinDanesz@users.noreply.github.com>
Date: Sun, 19 Jun 2022 23:03:07 +0200
Subject: [PATCH] fix: Fixed Containment spell for entities

---
 .../java/electroblob/wizardry/potion/PotionContainment.java  | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/src/main/java/electroblob/wizardry/potion/PotionContainment.java b/src/main/java/electroblob/wizardry/potion/PotionContainment.java
index cfd83c3a..2a7e5d9d 100644
--- a/src/main/java/electroblob/wizardry/potion/PotionContainment.java
+++ b/src/main/java/electroblob/wizardry/potion/PotionContainment.java
@@ -59,10 +59,9 @@ public class PotionContainment extends PotionMagicEffect {
 
 		if(x != target.posX || y != target.posY || z != target.posZ)
 		{
+			target.addVelocity(0.15 * Math.signum(x - target.posX), 0.15 * Math.signum(y - target.posY), 0.15 * Math.signum(z - target.posZ));
+			EntityUtils.undoGravity(target);
 			if(target.world.isRemote){
-				EntityUtils.undoGravity(target);
-				target.addVelocity(0.15 * Math.signum(x - target.posX), 0.15 * Math.signum(y - target.posY), 0.15 * Math.signum(z - target.posZ));
-
 				target.world.playSound(target.posX, target.posY, target.posZ, WizardrySounds.ENTITY_FORCEFIELD_DEFLECT,
 						WizardrySounds.SPELLS, 0.3f, 1f, false);
 			}