fixing the integration test