diff --git a/tensorflow_decision_forests/keras/BUILD b/tensorflow_decision_forests/keras/BUILD
index 613b3c2..fca67c8 100644
--- a/tensorflow_decision_forests/keras/BUILD
+++ b/tensorflow_decision_forests/keras/BUILD
@@ -1,5 +1,5 @@
-load("@rules_python//python:defs.bzl", "py_binary", "py_library")
 load("@org_tensorflow//tensorflow:tensorflow.bzl", "tf_cc_binary")
+load("@rules_python//python:defs.bzl", "py_binary", "py_library")
 load("//tensorflow_decision_forests/keras/wrapper:wrapper.bzl", "py_wrap_yggdrasil_learners")
 
 package(
diff --git a/tensorflow_decision_forests/tensorflow/ops/inference/BUILD b/tensorflow_decision_forests/tensorflow/ops/inference/BUILD
index 9efd78a..7d5fbbb 100644
--- a/tensorflow_decision_forests/tensorflow/ops/inference/BUILD
+++ b/tensorflow_decision_forests/tensorflow/ops/inference/BUILD
@@ -1,8 +1,8 @@
-load("@rules_python//python:defs.bzl", "py_library")
 load(
     "@org_tensorflow//tensorflow:tensorflow.bzl",
     "tf_gen_op_wrapper_py",
 )
+load("@rules_python//python:defs.bzl", "py_library")
 
 package(
     default_visibility = ["//visibility:public"],
diff --git a/tensorflow_decision_forests/tensorflow/ops/training/BUILD b/tensorflow_decision_forests/tensorflow/ops/training/BUILD
index 6c1e4b7..502c1b2 100644
--- a/tensorflow_decision_forests/tensorflow/ops/training/BUILD
+++ b/tensorflow_decision_forests/tensorflow/ops/training/BUILD
@@ -1,5 +1,5 @@
-load("@rules_python//python:defs.bzl", "py_library")
 load("@org_tensorflow//tensorflow:tensorflow.bzl", "tf_gen_op_wrapper_py")
+load("@rules_python//python:defs.bzl", "py_library")
 
 package(
     default_visibility = ["//visibility:public"],
diff --git a/tools/start_compile_docker.sh b/tools/start_compile_docker.sh
index 8710160..df67b72 100755
--- a/tools/start_compile_docker.sh
+++ b/tools/start_compile_docker.sh
@@ -58,7 +58,7 @@
 #  ./tools/build_pip_package.sh ALL_VERSIONS_ALREADY_ASSEMBLED
 #
 # https://hub.docker.com/r/tensorflow/build/tags?page=1
-DOCKER=tensorflow/build:2.16-python3.9
+DOCKER=tensorflow/build:2.17-python3.9
 
 # Current directory
 # Useful if Yggdrasil Decision Forests is available locally in a neighbor