Fix controller test