Added import to NAMESPACE to fix an error during gating.