Movement now works on the correct axis