diff --git a/pyproject.toml b/pyproject.toml
index 6adc673..6839cdd 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -22,7 +22,7 @@ authors = [
 ]
 
 dependencies = [
-  "segmentation-models-pytorch @ git+https://github.com/qubvel/segmentation_models.pytorch.git@master",
+  "segmentation-models-pytorch @ git+https://github.com/qubvel/segmentation_models.pytorch.git@main",
   "pandas>=1.5.3,<2.0",
   "scipy>=1.11.2,<2.0",
   "geo-benchmark>=0.0.3,<1.0",