diff --git a/tokens/_md-comp-checkbox.scss b/tokens/_md-comp-checkbox.scss
index c5593a5f9a..6305fd1ff8 100644
--- a/tokens/_md-comp-checkbox.scss
+++ b/tokens/_md-comp-checkbox.scss
@@ -4,11 +4,10 @@
 //
 
 // go/keep-sorted start
+@use './md-sys-color';
 @use './md-sys-shape';
 @use './md-sys-state';
-// TODO(b/272526637): tonal surface update
-@use './v0_161/md-comp-checkbox';
-@use './v0_161/md-sys-color';
+@use './v0_172/md-comp-checkbox';
 // go/keep-sorted end
 
 $_default: (