fixing the test for controller