Minor cleanup in integration test.