Update tests for controller