Fixed controller test code