From 09bcfd4fddccb026ea1e8319a5a98a8063a3f6cc Mon Sep 17 00:00:00 2001
From: MysticKoko <132279944+MysticKoko@users.noreply.github.com>
Date: Tue, 10 Dec 2024 19:31:29 +0200
Subject: [PATCH] Small tweak

---
 .../aquatic/entity/critter/HybridAquaticCritterEntity.kt      | 2 +-
 .../entity/crustacean/HybridAquaticCrustaceanEntity.kt        | 4 ++++
 .../dev/hybridlabs/aquatic/entity/miniboss/KarkinosEntity.kt  | 3 ++-
 3 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/src/main/kotlin/dev/hybridlabs/aquatic/entity/critter/HybridAquaticCritterEntity.kt b/src/main/kotlin/dev/hybridlabs/aquatic/entity/critter/HybridAquaticCritterEntity.kt
index 99ff8d5c7..9887809fc 100644
--- a/src/main/kotlin/dev/hybridlabs/aquatic/entity/critter/HybridAquaticCritterEntity.kt
+++ b/src/main/kotlin/dev/hybridlabs/aquatic/entity/critter/HybridAquaticCritterEntity.kt
@@ -62,7 +62,7 @@ open class HybridAquaticCritterEntity(
     }
 
     override fun hasNoDrag(): Boolean {
-        return false
+        return this.isSwimming
     }
 
     override fun tick() {
diff --git a/src/main/kotlin/dev/hybridlabs/aquatic/entity/crustacean/HybridAquaticCrustaceanEntity.kt b/src/main/kotlin/dev/hybridlabs/aquatic/entity/crustacean/HybridAquaticCrustaceanEntity.kt
index 4797ad753..9504b9711 100644
--- a/src/main/kotlin/dev/hybridlabs/aquatic/entity/crustacean/HybridAquaticCrustaceanEntity.kt
+++ b/src/main/kotlin/dev/hybridlabs/aquatic/entity/crustacean/HybridAquaticCrustaceanEntity.kt
@@ -267,6 +267,10 @@ open class HybridAquaticCrustaceanEntity(
         return false
     }
 
+    override fun hasNoDrag(): Boolean {
+        return this.isSwimming
+    }
+
     //#region SFX
 
     override fun calculateNextStepSoundDistance(): Float {
diff --git a/src/main/kotlin/dev/hybridlabs/aquatic/entity/miniboss/KarkinosEntity.kt b/src/main/kotlin/dev/hybridlabs/aquatic/entity/miniboss/KarkinosEntity.kt
index ce776a7aa..88f113b93 100644
--- a/src/main/kotlin/dev/hybridlabs/aquatic/entity/miniboss/KarkinosEntity.kt
+++ b/src/main/kotlin/dev/hybridlabs/aquatic/entity/miniboss/KarkinosEntity.kt
@@ -38,6 +38,7 @@ import software.bernie.geckolib.core.animation.AnimationController
 import software.bernie.geckolib.core.animation.RawAnimation
 import software.bernie.geckolib.core.`object`.PlayState
 
+
 class KarkinosEntity(entityType: EntityType<out HybridAquaticMinibossEntity>, world: World) :
     HybridAquaticMinibossEntity(entityType, world) {
 
@@ -87,7 +88,7 @@ class KarkinosEntity(entityType: EntityType<out HybridAquaticMinibossEntity>, wo
     }
 
     override fun hasNoDrag(): Boolean {
-        return false
+        return this.isSwimming
     }
 
     private fun getHandSwingDuration(): Int {