diff --git a/tpu_sync/frameworks/torch/BUILD b/tpu_sync/frameworks/torch/BUILD
index 5f21dab4..5f0e7dfd 100644
--- a/tpu_sync/frameworks/torch/BUILD
+++ b/tpu_sync/frameworks/torch/BUILD
@@ -429,6 +429,7 @@ cc_library(
     tags = ["nobuilder"],
     visibility = ["//visibility:public"],
     deps = [
+        ":torch_tpu_device_buffer_headers",
         ":torch_tpu_tensor_to_buffer_headers",
         ":torch_utils_mock",
         "//tpu_sync/weight_sync:weight_synchronizer_base",
diff --git a/tpu_sync/telemetry/prometheus_exporter.cc b/tpu_sync/telemetry/prometheus_exporter.cc
index 66882975..a62e9170 100644
--- a/tpu_sync/telemetry/prometheus_exporter.cc
+++ b/tpu_sync/telemetry/prometheus_exporter.cc
@@ -75,6 +75,7 @@ void PrometheusExporter::RegisterKnownFamilies() {
                             .Name(prometheus_name)
                             .Help(std::string(meta.description))
                             .Register(*registry_);
+        family->Add({});
         counter_families_.emplace(meta.name, family);
         break;
       }
@@ -83,6 +84,7 @@ void PrometheusExporter::RegisterKnownFamilies() {
                             .Name(prometheus_name)
                             .Help(std::string(meta.description))
                             .Register(*registry_);
+        family->Add({});
         gauge_families_.emplace(meta.name, family);
         break;
       }
@@ -91,6 +93,7 @@ void PrometheusExporter::RegisterKnownFamilies() {
                             .Name(prometheus_name)
                             .Help(std::string(meta.description))
                             .Register(*registry_);
+        family->Add({}, default_buckets_);
         histogram_families_.emplace(meta.name, family);
         break;
       }