diff --git a/pvnet/data/site_datamodule.py b/pvnet/data/site_datamodule.py
index b55803ec..41261c59 100644
--- a/pvnet/data/site_datamodule.py
+++ b/pvnet/data/site_datamodule.py
@@ -2,7 +2,7 @@
 from glob import glob
 
 import xarray as xr
-from ocf_data_sampler.torch_datasets.site import SitesDataset, convert_netcdf_to_numpy_sample
+from ocf_data_sampler.torch_datasets.datasets.site import SitesDataset, convert_netcdf_to_numpy_sample
 from torch.utils.data import Dataset
 
 from pvnet.data.base_datamodule import BaseDataModule
diff --git a/pvnet/data/uk_regional_datamodule.py b/pvnet/data/uk_regional_datamodule.py
index 725243fc..59d78011 100644
--- a/pvnet/data/uk_regional_datamodule.py
+++ b/pvnet/data/uk_regional_datamodule.py
@@ -3,7 +3,7 @@
 
 import torch
 from pvnet.data.base_datamodule import BaseDataModule
-from ocf_data_sampler.torch_datasets.pvnet_uk_regional import PVNetUKRegionalDataset
+from ocf_data_sampler.torch_datasets.datasets.pvnet_uk_regional import PVNetUKRegionalDataset
 from torch.utils.data import Dataset
 
 
diff --git a/pyproject.toml b/pyproject.toml
index 1fd0e2c4..db41d3fd 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -6,7 +6,7 @@ dynamic = ["version", "readme"]
 license={file="LICENCE"}
 
 dependencies = [
-    "ocf_data_sampler==0.0.43",
+    "ocf_data_sampler==0.0.47",
     "ocf_datapipes>=3.3.34",
     "ocf_ml_metrics>=0.0.11",
     "numpy",