fixed the controller unit test