Unit bugfix