Есть ещё один визуальный баг. При движении влево-вправо состояния ходьбы и полёта сменяются очень быстро. Это происходит потому что сила притяжения сильнее силы полёта и постоянно тянет моль вниз. Надо либо увеличить силу полёта либо уменьшить гравитацию моли во время полёта. Логично было бы сделать первое, но мы сделаем второе.

this.onStateEnter("fly", () => {
this.play("fly")
this.gravityScale = 0
})

this.onStateEnter("idle", () => {
this.play("idle")
this.gravityScale = 1
})

joystick.on("endMove", () => {
this.gravityScale = 1
if(this.isGrounded()) this.enterState("idle")
})

Просто ставим умножение гравитации на ноль во время полёта и возвращаем до единицы при приземлении. Возможно это плохое решение и мы его изменим в будущем, но а сейчас оно работает так как нам надо)