fixed mising unit
Merge pull request #11 from Korrigan/master

fixed mising unit