diff --git a/icenet/data/dataset.py b/icenet/data/dataset.py
index 0563684..511979c 100644
--- a/icenet/data/dataset.py
+++ b/icenet/data/dataset.py
@@ -370,9 +370,9 @@ def __getitem__(self, idx):
         def dates(self):
             return self._dates
 except ModuleNotFoundError:
-    logging.warn("PyTorch module not found - not mandatory if not using PyTorch")
+    logging.warning("PyTorch module not found - not mandatory if not using PyTorch")
 except ImportError:
-    logging.warn("PyTorch import failed - not mandatory if not using PyTorch")
+    logging.warning("PyTorch import failed - not mandatory if not using PyTorch")
 
 
 @setup_logging