diff --git a/CHANGELOG.md b/CHANGELOG.md
index 97ea8098..c9f1a4f2 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -1,5 +1,11 @@
 # Changelog
 
+## 0.1.1 - 2021-05-17
+
+### Features
+
+-   Migration to TensorFlow 2.5.0.
+
 ## 0.1.0 - 2021-05-11
 
 Initial release of Yggdrasil Decision Forests.
diff --git a/documentation/installation.md b/documentation/installation.md
index a4a50104..f524d66b 100644
--- a/documentation/installation.md
+++ b/documentation/installation.md
@@ -60,7 +60,7 @@ Once Bazel is installed, clone the github repository and start the compilation:
 
 ```shell
 git clone https://github.com/google/yggdrasil-decision-forests.git
-cd yggdrasil_decision_forests
+cd yggdrasil-decision-forests
 
 bazel build //yggdrasil_decision_forests/...:all --config=<platform config>
 ```
@@ -71,7 +71,7 @@ For example:
 
 ```shell
 git clone https://github.com/google/yggdrasil-decision-forests.git
-cd yggdrasil_decision_forests
+cd yggdrasil-decision-forests
 
 bazel build //yggdrasil_decision_forests/cli/...:all --config=linux_cpp17 --config=linux_avx2
 ```
@@ -97,7 +97,7 @@ pip is not available on the paths used by bazel.
 set PYTHON_BIN_PATH=C:\Python38\python.exe
 
 git clone https://github.com/google/yggdrasil-decision-forests.git
-cd yggdrasil_decision_forests
+cd yggdrasil-decision-forests
 
 bazel build //yggdrasil_decision_forests/cli/...:all --config=windows_cpp17 --config=windows_avx2
 ```
diff --git a/third_party/tensorflow/workspace.bzl b/third_party/tensorflow/workspace.bzl
index f3861c57..a9539077 100644
--- a/third_party/tensorflow/workspace.bzl
+++ b/third_party/tensorflow/workspace.bzl
@@ -5,15 +5,7 @@ load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
 def deps():
     http_archive(
         name = "org_tensorflow",
-        # 2.4.1
-        #sha256 = "b91ec194ddf6c4a5a2f9d1db4af4daab0b187ff691e6f88142413d2c7e77a3bb",
-        #strip_prefix = "tensorflow-2.4.1",
-        #urls = ["https://github.com/tensorflow/tensorflow/archive/v2.4.1.zip"],
-        # v2.5.0-rcx
-        sha256 = "f7ad0a488559ee01f042a967c065482b074e11afd0299facbd8dc0cba9ae3fa9",
-        strip_prefix = "tensorflow-2.5.0-rc3",
-        urls = ["https://github.com/tensorflow/tensorflow/archive/refs/tags/v2.5.0-rc3.zip"],
-        # head
-        #urls = ["https://github.com/tensorflow/tensorflow/archive/master.zip"],
-        #strip_prefix = "tensorflow-master",
+        sha256 = "e3d0ee227cc19bd0fa34a4539c8a540b40f937e561b4580d4bbb7f0e31c6a713",
+        strip_prefix = "tensorflow-2.5.0",
+        urls = ["https://github.com/tensorflow/tensorflow/archive/refs/tags/v2.5.0.zip"],
     )