diff --git a/modules/jni/src/main/scala/org/platanios/tensorflow/jni/TensorFlow.scala b/modules/jni/src/main/scala/org/platanios/tensorflow/jni/TensorFlow.scala
index 458761701..934da192d 100644
--- a/modules/jni/src/main/scala/org/platanios/tensorflow/jni/TensorFlow.scala
+++ b/modules/jni/src/main/scala/org/platanios/tensorflow/jni/TensorFlow.scala
@@ -123,10 +123,10 @@ object TensorFlow {
       Seq(
         (s"lib$lib.so.link", false),
         (s"lib$lib.so.2.link", false),
-        (s"lib$lib.so.2.3.0", false),
+        (s"lib$lib.so.2.3.1", false),
         (s"lib$lib.dylib.link", false),
         (s"lib$lib.2.dylib.link", false),
-        (s"lib$lib.2.3.0.dylib", false),
+        (s"lib$lib.2.3.1.dylib", false),
       )
     }
   }
diff --git a/version.sbt b/version.sbt
index c9c37f00d..9aefe74c4 100644
--- a/version.sbt
+++ b/version.sbt
@@ -1 +1 @@
-version in ThisBuild := "0.5.6-SNAPSHOT"
+version in ThisBuild := "0.5.6"