diff --git a/setup.py b/setup.py
index 7195da6..16263c4 100644
--- a/setup.py
+++ b/setup.py
@@ -25,7 +25,7 @@ def read(fname):
 
 setup(
     name="traintrack",
-    version="0.1.4",
+    version="0.1.6",
     description="A simple helper to run pipelines of PytorchLightning models",
     author="Daniel Murnane",
     install_requires=dependencies,