fix import in documentation