Fix error on single unit specification