diff --git a/setup.py b/setup.py
index 5939c67..0981222 100644
--- a/setup.py
+++ b/setup.py
@@ -7,7 +7,7 @@
 if __name__ == "__main__":
     setup(
         name="tez",
-        version="0.1.6",
+        version="0.1.7",
         description="tez - a simple pytorch trainer",
         long_description=long_description,
         long_description_content_type="text/markdown",