fix updateUnitLocations
Merge pull request #3 from jessetane/fix-unit-locations

Fix unit locations