diff --git a/src/generators/mkui.ts b/src/generators/mkui.ts
index 2dc58e8..9e669f6 100644
--- a/src/generators/mkui.ts
+++ b/src/generators/mkui.ts
@@ -236,6 +236,10 @@ function mkui({ ui, code }: MKUIParams): UIColourPalette {
     'activityBarTop.activeBackground': ui.background.panel.hex(),
     'activityBarBadge.background': ui.accent.hex(),
     'activityBarBadge.foreground': ui.text.contrast.hex(),
+    'activityErrorBadge.background': ui.problems.error.hex(),
+    'activityErrorBadge.foreground': ui.text.contrast.hex(),
+    'activityWarningBadge.background': ui.problems.warning.hex(),
+    'activityWarningBadge.foreground': ui.text.contrast.hex(),
     // TODO(vscode): implement panelTitle.activeBackground colour contribution
     // 'activityBar.activeBackground': ui.background.code.hex(),