minor updates to the integration test.