Update controller test
Update controller test