integration test improvement