Fix unit movement regression

Fix a regression where the unit would no longer keep track of how far it
has moved.