diff --git a/src/webviews/apps/home/home.ts b/src/webviews/apps/home/home.ts
index 6546259..210ed6c 100644
--- a/src/webviews/apps/home/home.ts
+++ b/src/webviews/apps/home/home.ts
@@ -17,11 +17,7 @@ export class HomeApp extends App<State> {
 	}
 
 	protected override onInitialize() {
-		provideVSCodeDesignSystem().register({
-			register: function (container: any, context: any) {
-				vsCodeButton().register(container, context);
-			},
-		});
+		provideVSCodeDesignSystem().register(vsCodeButton());
 
 		this.$slots = [
 			document.getElementById('slot1') as HTMLDivElement,
diff --git a/src/webviews/apps/plus/timeline/timeline.ts b/src/webviews/apps/plus/timeline/timeline.ts
index d0bdb0d..eb2a25f 100644
--- a/src/webviews/apps/plus/timeline/timeline.ts
+++ b/src/webviews/apps/plus/timeline/timeline.ts
@@ -22,13 +22,7 @@ export class TimelineApp extends App<State> {
 	}
 
 	protected override onInitialize() {
-		provideVSCodeDesignSystem().register({
-			register: function (container: any, context: any) {
-				vsCodeButton().register(container, context);
-				vsCodeDropdown().register(container, context);
-				vsCodeOption().register(container, context);
-			},
-		});
+		provideVSCodeDesignSystem().register(vsCodeButton(), vsCodeDropdown(), vsCodeOption());
 
 		this.updateState();
 	}