correct import in docs