Fixed import in integration tests