Fix wrong import in doc