Small update to the controller test file.