diff --git a/fabric/src/main/java/me/lambdaurora/lambdacontrols/client/mixin/ClientPlayerEntityMixin.java b/fabric/src/main/java/me/lambdaurora/lambdacontrols/client/mixin/ClientPlayerEntityMixin.java
index d694432..cd41081 100644
--- a/fabric/src/main/java/me/lambdaurora/lambdacontrols/client/mixin/ClientPlayerEntityMixin.java
+++ b/fabric/src/main/java/me/lambdaurora/lambdacontrols/client/mixin/ClientPlayerEntityMixin.java
@@ -19,7 +19,8 @@ import net.minecraft.util.math.Vec3d;
 import org.spongepowered.asm.mixin.Mixin;
 import org.spongepowered.asm.mixin.Shadow;
 import org.spongepowered.asm.mixin.injection.At;
-import org.spongepowered.asm.mixin.injection.Redirect;
+import org.spongepowered.asm.mixin.injection.Inject;
+import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
 
 /**
  * Injects the anti fly drifting feature.
@@ -38,12 +39,12 @@ public abstract class ClientPlayerEntityMixin extends AbstractClientPlayerEntity
         super(world, profile);
     }
 
-    @Redirect(method = "move(Lnet/minecraft/entity/MovementType;Lnet/minecraft/util/math/Vec3d;)V", at = @At(value = "INVOKE", target = "Lnet/minecraft/client/network/AbstractClientPlayerEntity;move(Lnet/minecraft/entity/MovementType;Lnet/minecraft/util/math/Vec3d;)V"))
-    public void lambdacontrols_move(AbstractClientPlayerEntity player, MovementType type, Vec3d movement)
+    @Inject(method = "move(Lnet/minecraft/entity/MovementType;Lnet/minecraft/util/math/Vec3d;)V", at = @At(value = "INVOKE", target = "Lnet/minecraft/client/network/AbstractClientPlayerEntity;move(Lnet/minecraft/entity/MovementType;Lnet/minecraft/util/math/Vec3d;)V"))
+    public void lambdacontrols_move(MovementType type, Vec3d movement, CallbackInfo ci)
     {
         LambdaControlsClient mod = LambdaControlsClient.get();
         if (type == MovementType.SELF) {
-            if (player.abilities.flying && !mod.config.has_fly_drifting()) {
+            if (this.abilities.flying && !mod.config.has_fly_drifting()) {
                 if (!this.method_22120()) {
                     if (!this.lambdacontrols_drifting_prevented) {
                         this.setVelocity(this.getVelocity().multiply(0, 1.0, 0));
@@ -53,6 +54,5 @@ public abstract class ClientPlayerEntityMixin extends AbstractClientPlayerEntity
                     this.lambdacontrols_drifting_prevented = false;
             }
         }
-        super.move(type, movement);
     }
 }