Fix diff drive unit test

Previous commits left the diff drive unit tests broken. This is a fix for them.
Proper unit tests will be done for pending pull requests.