diff --git a/packages/blocks/src/code-block/code-block-service.ts b/packages/blocks/src/code-block/code-block-service.ts
index e9d91453414b..fb57f8462a58 100644
--- a/packages/blocks/src/code-block/code-block-service.ts
+++ b/packages/blocks/src/code-block/code-block-service.ts
@@ -30,7 +30,7 @@ export class CodeBlockService extends BlockService {
   }
 
   get themeKey() {
-    const theme = this.std.get(ThemeProvider).theme;
+    const theme = this.std.get(ThemeProvider).theme$.value;
     return theme === ColorScheme.Dark
       ? this._darkThemeKey
       : this._lightThemeKey;