diff --git a/keras_contrib/layers/crf.py b/keras_contrib/layers/crf.py
index 88a64ac69..e48db76a1 100644
--- a/keras_contrib/layers/crf.py
+++ b/keras_contrib/layers/crf.py
@@ -513,7 +513,7 @@ def recursion(self, input_energy, mask=None, go_backwards=False,
         constants = [chain_energy]
 
         if mask is not None:
-            mask2 = K.cast(K.concatenate([mask, K.zeros_like(mask[:, :1])], axis=1),
+            mask2 = K.cast(K.concatenate([mask, K.cast(K.zeros_like(mask[:, :1]), mask.dtype)], axis=1),
                            K.floatx())
             constants.append(mask2)
 
diff --git a/setup.py b/setup.py
index 569e5cae5..d5a508ac2 100644
--- a/setup.py
+++ b/setup.py
@@ -11,7 +11,7 @@
     install_requires = ['keras']
 
 setup(name=name,
-      version='2.0.8',
+      version='2.0.8.1',
       description='Keras Deep Learning for Python, Community Contributions',
       author='Fariz Rahman',
       author_email='farizrahman4u@gmail.com',