fix unit movement