diff --git a/neural_structured_learning/BUILD b/neural_structured_learning/BUILD
index eda5fd8..23a4fec 100644
--- a/neural_structured_learning/BUILD
+++ b/neural_structured_learning/BUILD
@@ -15,8 +15,8 @@
 # Description:
 # Build rules for TensorFlow Neural Structured Learning.
 
-# Placeholder for internal Python strict compatibility macro.
 # Internal annotation for sync
+# Placeholder for internal Python strict compatibility macro.
 load("@rules_license//rules:license.bzl", "license")
 
 package(
diff --git a/research/carls/knowledge_bank/initializer_helper.cc b/research/carls/knowledge_bank/initializer_helper.cc
index 6bab3f2..98d0691 100644
--- a/research/carls/knowledge_bank/initializer_helper.cc
+++ b/research/carls/knowledge_bank/initializer_helper.cc
@@ -68,7 +68,7 @@ EmbeddingVectorProto InitializeEmbedding(
     return result;
   }
   if (initializer.has_random_uniform_initializer()) {
-    absl::BitGen bit_gen;
+    absl::SharedBitGen bit_gen;
     const auto& init = initializer.random_uniform_initializer();
     for (int i = 0; i < embedding_dimension; ++i) {
       result.add_value(absl::Uniform<float>(bit_gen, init.low(), init.high()));
@@ -76,7 +76,7 @@ EmbeddingVectorProto InitializeEmbedding(
     return result;
   }
   if (initializer.has_random_normal_initializer()) {
-    absl::BitGen bit_gen;
+    absl::SharedBitGen bit_gen;
     const auto& init = initializer.random_normal_initializer();
     for (int i = 0; i < embedding_dimension; ++i) {
       result.add_value(