Updating integration test to check for appropriate keys in returned value