fix unit test controller