small fix to unit