diff --git a/setup.cfg b/setup.cfg
index 3ef6a96..87c0b6f 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 0.17.0
+current_version = 0.17.1
 commit = True
 tag = True
 
diff --git a/setup.py b/setup.py
index 704cc9c..3fdd1ac 100644
--- a/setup.py
+++ b/setup.py
@@ -38,6 +38,6 @@
     setup_requires=setup_requirements,
     tests_require=test_requirements,
     test_suite='tests',
-    version='0.17.0',
+    version='0.17.1',
     zip_safe=False,
 )
diff --git a/torchkge/__init__.py b/torchkge/__init__.py
index 68364a8..0a72e44 100755
--- a/torchkge/__init__.py
+++ b/torchkge/__init__.py
@@ -4,7 +4,7 @@
 
 __author__ = """Armand Boschin"""
 __email__ = 'aboschin@enst.fr'
-__version__ = '0.17.0'
+__version__ = '0.17.1'
 
 from torchkge.exceptions import NotYetEvaluatedError
 from torchkge.utils import MarginLoss, LogisticLoss