integration test fix