Minor modification to integration test