Update to test controller