diff --git a/tensorboard/plugins/interactive_inference/tf_interactive_inference_dashboard/tf-interactive-inference-dashboard.html b/tensorboard/plugins/interactive_inference/tf_interactive_inference_dashboard/tf-interactive-inference-dashboard.html
index 3dd8b73101..0552935ca4 100644
--- a/tensorboard/plugins/interactive_inference/tf_interactive_inference_dashboard/tf-interactive-inference-dashboard.html
+++ b/tensorboard/plugins/interactive_inference/tf_interactive_inference_dashboard/tf-interactive-inference-dashboard.html
@@ -6740,12 +6740,13 @@ <h2>Show similarity to selected datapoint</h2>
         calculateBucketEdges_: function(feature, numBuckets) {
           if (
             !this.isNumericFeature_(feature) ||
-            // No point in aggregating if less unique values than buckets.
-            this.stats[feature].uniqueCount < numBuckets ||
+            // No point in aggregating if not more unique values than buckets.
+            this.stats[feature].uniqueCount <= numBuckets ||
             // Already done.
             (this.featureBucketEdges_[feature] &&
               this.featureBucketEdges_[feature].length == numBuckets + 1)
           ) {
+            delete this.featureBucketEdges_[feature];
             return;
           }
           const min = this.stats[feature].numberMin;