Modified integration test