Fix integration test helper