Fixed bug in unit object editing

so ein schöner pull request <3