Updated the test controller to the latest code.