Fixed the controller test