diff --git a/setup.py b/setup.py
index 0858c1f..19efcf0 100644
--- a/setup.py
+++ b/setup.py
@@ -13,6 +13,6 @@
           'scikit-image',
           'torch',
           'torchvision',
-          'lungmask @ git+https://git@github.com/JoHof/lungmask@master',
+          'lungmask @ git+https://git@github.com/JoHof/lungmask@v0.2.4',
       ],
        entry_points ={'console_scripts': ['balaitous = balaitous.cli:cli']})