Fix a couple of bugs

Author: Earthcomputer

gradle.properties

@@ -7,7 +7,7 @@ org.gradle.jvmargs=-Xmx1G
 	loader_version=0.8.2+build.194
 
 # Mod Properties
-	mod_version = 1.4
+	mod_version = 1.4.1
 	maven_group = net.earthcomputer
 	archives_base_name = movement-hacks
 

src/main/java/net/earthcomputer/movementhacks/mixin/EntityAccessor.java

@@ -0,0 +1,11 @@
+package net.earthcomputer.movementhacks.mixin;
+
+import net.minecraft.src.Entity;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.gen.Accessor;
+
+@Mixin(Entity.class)
+public interface EntityAccessor {
+    @Accessor
+    void setField_653_b(int field_653_b);
+}

src/main/java/net/earthcomputer/movementhacks/mixin/MixinEntityPlayerSP.java

@@ -1,10 +1,7 @@
 package net.earthcomputer.movementhacks.mixin;
 
 import net.earthcomputer.movementhacks.IPlayer;
-import net.minecraft.src.EntityPlayer;
-import net.minecraft.src.EntityPlayerSP;
-import net.minecraft.src.MovementInput;
-import net.minecraft.src.World;
+import net.minecraft.src.*;
 import org.spongepowered.asm.mixin.Mixin;
 import org.spongepowered.asm.mixin.Shadow;
 import org.spongepowered.asm.mixin.Unique;
@@ -32,8 +29,13 @@ public boolean isFlying() {
     @Override
     public void setFlying(boolean flying) {
         this.flying = flying;
-        if (flying)
+        if (flying) {
             immuneToFallDamage = true;
+        } else {
+            // step sound counter
+            ((EntityAccessor) this).setField_653_b(1);
+            this.field_641_aF = 0;
+        }
     }
 
     @Override
@@ -68,4 +70,14 @@ private void onTick(CallbackInfo ci) {
         if (onGround)
             flying = immuneToFallDamage = false;
     }
+
+    @Inject(method = "writeEntityToNBT", at = @At("RETURN"))
+    private void onWriteEntityToNBT(NBTTagCompound nbt, CallbackInfo ci) {
+        nbt.setBoolean("Flying", flying);
+    }
+
+    @Inject(method = "readEntityFromNBT", at = @At("RETURN"))
+    private void onReadEntityFromNBT(NBTTagCompound nbt, CallbackInfo ci) {
+        setFlying(nbt.getBoolean("Flying"));
+    }
 }

src/main/resources/movementhacks.mixins.json

@@ -3,12 +3,13 @@
   "package": "net.earthcomputer.movementhacks.mixin",
   "compatibilityLevel": "JAVA_8",
   "mixins": [
+    "EntityAccessor"
   ],
   "client": [
-    "MixinEntityPlayerSP",
-    "MixinMovementInputFromOptions",
+    "MixinEntity",
     "MixinEntityLiving",
-    "MixinEntity"
+    "MixinEntityPlayerSP",
+    "MixinMovementInputFromOptions"
   ],
   "injectors": {
     "defaultRequire": 1