Minor update to the controller test.