Integration test fix