Add missing end to work units controller spec