diff --git a/python/python/lance/lance/__init__.pyi b/python/python/lance/lance/__init__.pyi
index 6e5e40fc23b..ff8d5e4e4d5 100644
--- a/python/python/lance/lance/__init__.pyi
+++ b/python/python/lance/lance/__init__.pyi
@@ -582,5 +582,14 @@ class ScanStatistics:
         str, int
     ]  # Additional metrics for debugging purposes. Subject to change.
 
+class DatasetBasePath:
+    def __init__(
+        self,
+        path: str,
+        name: Optional[str] = None,
+        is_dataset_root: bool = False,
+        id: Optional[int] = None,
+    ) -> None: ...
+
 __version__: str
 language_model_home: Callable[[], str]
diff --git a/python/src/dataset.rs b/python/src/dataset.rs
index beb8a06537d..68031cc3c70 100644
--- a/python/src/dataset.rs
+++ b/python/src/dataset.rs
@@ -389,7 +389,7 @@ impl<'py> IntoPyObject<'py> for PyLance<&ColumnOrdering> {
 }
 
 /// Python binding for BasePath
-#[pyclass(name = "DatasetBasePath", module = "lance")]
+#[pyclass(name = "DatasetBasePath", module = "_lib")]
 #[derive(Clone)]
 pub struct DatasetBasePath {
     #[pyo3(get)]