Fix the unit name (whoops)