Change the logic for movement; this way seems to be closer to how it works in Hero Mesh, due to my experiments