diff --git a/setup.py b/setup.py
index ebc72e4..a69398c 100644
--- a/setup.py
+++ b/setup.py
@@ -14,7 +14,7 @@
 
 setup(
     name='keras-tcn',
-    version='3.5.0',
+    version='3.5.1',
     description='Keras TCN',
     author='Philippe Remy',
     license='MIT',
diff --git a/tcn/__init__.py b/tcn/__init__.py
index cd15d82..17556db 100644
--- a/tcn/__init__.py
+++ b/tcn/__init__.py
@@ -1,3 +1,3 @@
 from tcn.tcn import TCN, compiled_tcn, tcn_full_summary  # noqa
 
-__version__ = '3.5.0'
+__version__ = '3.5.1'