diff --git a/tensorboard/webapp/theme/_tb_theme.template.scss b/tensorboard/webapp/theme/_tb_theme.template.scss
index 2e4f820e54..7ccd26771c 100644
--- a/tensorboard/webapp/theme/_tb_theme.template.scss
+++ b/tensorboard/webapp/theme/_tb_theme.template.scss
@@ -61,7 +61,7 @@ $tb-foreground: map_merge(
   )
 );
 $tb-background: map_merge(
-  matm2.$light-theme-background-palette,
+  matm2var.$light-theme-background-palette,
   (
     app-bar: $tb-app-bar-color,
     // Default is `map.get($grey-palette, 50)`.
diff --git a/tensorboard/webapp/widgets/data_table/column_selector_component.scss b/tensorboard/webapp/widgets/data_table/column_selector_component.scss
index 1e82a25b31..3bd06b62a0 100644
--- a/tensorboard/webapp/widgets/data_table/column_selector_component.scss
+++ b/tensorboard/webapp/widgets/data_table/column_selector_component.scss
@@ -82,10 +82,16 @@ limitations under the License.
     width: 100%;
 
     &.selected {
-      background-color: matm2.get-color-from-palette(mat.$grey-palette, 200);
+      background-color: matm2.get-color-from-palette(
+        matm2var.$grey-palette,
+        200
+      );
 
       @include tb-dark-theme {
-        background-color: matm2.get-color-from-palette(mat.$grey-palette, 400);
+        background-color: matm2.get-color-from-palette(
+          matm2var.$grey-palette,
+          400
+        );
       }
     }
   }