diff --git a/EasyMinecraftGoals/v1_21_R1/src/main/java/com/magmaguy/easyminecraftgoals/v1_21_R1/hitbox/Hitbox.java b/EasyMinecraftGoals/v1_21_R1/src/main/java/com/magmaguy/easyminecraftgoals/v1_21_R1/hitbox/Hitbox.java
index fa1ef93..ee32edb 100644
--- a/EasyMinecraftGoals/v1_21_R1/src/main/java/com/magmaguy/easyminecraftgoals/v1_21_R1/hitbox/Hitbox.java
+++ b/EasyMinecraftGoals/v1_21_R1/src/main/java/com/magmaguy/easyminecraftgoals/v1_21_R1/hitbox/Hitbox.java
@@ -2,6 +2,7 @@
 
 import net.minecraft.world.entity.Entity;
 import net.minecraft.world.entity.EntityDimensions;
+import net.minecraft.world.phys.AABB;
 
 import java.lang.reflect.Field;
 
@@ -11,11 +12,13 @@ private Hitbox() {
 
     public static boolean setCustomHitbox(Entity entity, float width, float height, boolean fixed) {
         EntityDimensions entityDimensions = new EntityDimensions(width, height, height, null, fixed);
+        AABB boundingBox = entityDimensions.makeBoundingBox(entity.position());
+
         Class<?> entityClass = Entity.class;
         try {
             Field field = entityClass.getDeclaredField("aF");
             field.setAccessible(true);
-            field.set(entity, entityDimensions);
+            field.set(entity, boundingBox);
         } catch (NoSuchFieldException e) {
             e.printStackTrace();
             return false;
@@ -23,7 +26,7 @@ public static boolean setCustomHitbox(Entity entity, float width, float height,
             e.printStackTrace();
             return false;
         }
-        entity.setBoundingBox(entityDimensions.makeBoundingBox(entity.position()));
+        entity.setBoundingBox(boundingBox);
         return true;
     }
 }