Update test for controller