Updating integration test