diff --git a/packages/material-ui-lab/src/ToggleButton/ToggleButton.d.ts b/packages/material-ui-lab/src/ToggleButton/ToggleButton.d.ts
index c03f510e4abcce..3e99855b647c61 100644
--- a/packages/material-ui-lab/src/ToggleButton/ToggleButton.d.ts
+++ b/packages/material-ui-lab/src/ToggleButton/ToggleButton.d.ts
@@ -13,6 +13,12 @@ declare const ToggleButton: ExtendButtonBase<{
 
 export type ToggleButtonProps = SimplifiedPropsOf<typeof ToggleButton>;
 
-export type ToggleButtonClassKey = ButtonBaseClassKey | 'label' | 'selected';
+export type ToggleButtonClassKey =
+  | 'root'
+  | 'disabled'
+  | 'selected'
+  | 'label'
+  | 'sizeSmall'
+  | 'sizeLarge';
 
 export default ToggleButton;
diff --git a/packages/material-ui-lab/src/ToggleButtonGroup/ToggleButtonGroup.d.ts b/packages/material-ui-lab/src/ToggleButtonGroup/ToggleButtonGroup.d.ts
index 0501710bf5eecf..95b1fdeaa17c5a 100644
--- a/packages/material-ui-lab/src/ToggleButtonGroup/ToggleButtonGroup.d.ts
+++ b/packages/material-ui-lab/src/ToggleButtonGroup/ToggleButtonGroup.d.ts
@@ -16,7 +16,11 @@ export interface ToggleButtonGroupProps
   value?: any;
 }
 
-export type ToggleButtonGroupClassKey = 'root' | 'selected';
+export type ToggleButtonGroupClassKey =
+  | 'root'
+  | 'grouped'
+  | 'groupedSizeSmall'
+  | 'groupedSizeLarge';
 
 declare const ToggleButtonGroup: React.ComponentType<ToggleButtonGroupProps>;