Integrations test fix