diff --git a/common/src/main/java/net/cmr/jurassicrevived/entity/client/AlligatorGarModel.java b/common/src/main/java/net/cmr/jurassicrevived/entity/client/AlligatorGarModel.java
index 70ec1dc..aad448b 100644
--- a/common/src/main/java/net/cmr/jurassicrevived/entity/client/AlligatorGarModel.java
+++ b/common/src/main/java/net/cmr/jurassicrevived/entity/client/AlligatorGarModel.java
@@ -6,18 +6,17 @@ import net.cmr.jurassicrevived.entity.custom.AlligatorGarEntity;
 import net.minecraft.Util;
 import net.minecraft.resources.ResourceLocation;
 import net.minecraft.util.Mth;
+/*? if <=1.20.1 {*/
+import software.bernie.geckolib.core.animation.AnimationState;
+/*?} else {*/
+/*import software.bernie.geckolib.animation.AnimationState;
+ *//*?}*/
 import software.bernie.geckolib.cache.object.GeoBone;
 import software.bernie.geckolib.constant.DataTickets;
-import software.bernie.geckolib.core.animation.AnimationState;
 import software.bernie.geckolib.model.GeoModel;
 
 import java.util.Map;
 
-/*? if <=1.20.1 {*/
-/*?} else {*/
-/*import software.bernie.geckolib.animation.AnimationState;
-*//*?}*/
-
 public class AlligatorGarModel extends GeoModel<AlligatorGarEntity> {
 
     private static final Map<AlligatorGarVariant, ResourceLocation> LOCATION_BY_VARIANT =
diff --git a/common/src/main/java/net/cmr/jurassicrevived/entity/client/MawsoniaModel.java b/common/src/main/java/net/cmr/jurassicrevived/entity/client/MawsoniaModel.java
index e3a88be..c9d517b 100644
--- a/common/src/main/java/net/cmr/jurassicrevived/entity/client/MawsoniaModel.java
+++ b/common/src/main/java/net/cmr/jurassicrevived/entity/client/MawsoniaModel.java
@@ -6,19 +6,17 @@ import net.cmr.jurassicrevived.entity.custom.MawsoniaEntity;
 import net.minecraft.Util;
 import net.minecraft.resources.ResourceLocation;
 import net.minecraft.util.Mth;
+/*? if <=1.20.1 {*/
+import software.bernie.geckolib.core.animation.AnimationState;
+/*?} else {*/
+/*import software.bernie.geckolib.animation.AnimationState;
+ *//*?}*/
 import software.bernie.geckolib.cache.object.GeoBone;
 import software.bernie.geckolib.constant.DataTickets;
-import software.bernie.geckolib.core.animation.AnimationState;
 import software.bernie.geckolib.model.GeoModel;
 
 import java.util.Map;
 
-/*? if <=1.20.1 {*/
-/*?} else {*/
-/*import software.bernie.geckolib.animation.AnimationState;
- 
-*//*?}*/
-
 public class MawsoniaModel extends GeoModel<MawsoniaEntity> {
 
     private static final Map<MawsoniaVariant, ResourceLocation> LOCATION_BY_VARIANT =