Updated tests to check for expected response output