Integration test fix.