diff --git a/tensorboard/components/vz_example_viewer/vz-example-viewer.ts b/tensorboard/components/vz_example_viewer/vz-example-viewer.ts
index 13b04dae26..26b2a2db1e 100644
--- a/tensorboard/components/vz_example_viewer/vz-example-viewer.ts
+++ b/tensorboard/components/vz_example_viewer/vz-example-viewer.ts
@@ -922,9 +922,9 @@ Polymer({
     let jsonFeat: any;
     if (this.newFeatureType === INT_FEATURE_NAME) {
       const valueList: number[] = [];
-      const ints = new FloatList();
+      const ints = new Int64List();
       ints.setValueList(valueList);
-      feat.setInt64List(ints as any);  // TODO(#2266): Preserves behavior.
+      feat.setInt64List(ints);
       jsonFeat = {int64List: {value: valueList}};
     } else if (this.newFeatureType === FLOAT_FEATURE_NAME) {
       const valueList: number[] = [];