Update tests for controllers