Fixed controller test to work properly