fix import in docs