Fixed bug in unit object editing
Merge pull request #281 from Crigges/master

Fixed bug in unit object editing