Fix movement :)