diff --git a/packages/mui-codemod/src/deprecations/badge-props/badge-props.test.js b/packages/mui-codemod/src/deprecations/badge-props/badge-props.test.js
index 0816df85024551..d1426f7a95f1e7 100644
--- a/packages/mui-codemod/src/deprecations/badge-props/badge-props.test.js
+++ b/packages/mui-codemod/src/deprecations/badge-props/badge-props.test.js
@@ -12,18 +12,14 @@ describe('@mui/codemod', () => {
   describe('deprecations', () => {
     describe('badge-props', () => {
       it('transforms props as needed', () => {
-        const actual = transform(
-          { source: read('./test-cases/actual.js') },
-          { jscodeshift },
-          { printOptions: { trailingComma: true } },
-        );
+        const actual = transform({ source: read('./test-cases/actual.js') }, { jscodeshift }, {});
 
         const expected = read('./test-cases/expected.js');
         expect(actual).to.equal(expected, 'The transformed version should be correct');
       });
 
       it('should be idempotent', () => {
-        const actual = transform({ source: read('./test-cases/actual.js') }, { jscodeshift }, {});
+        const actual = transform({ source: read('./test-cases/expected.js') }, { jscodeshift }, {});
 
         const excepted = read('./test-cases/expected.js');
         expect(actual).to.equal(excepted, 'The transformed version should be correct');
diff --git a/packages/mui-codemod/src/deprecations/badge-props/test-cases/expected.js b/packages/mui-codemod/src/deprecations/badge-props/test-cases/expected.js
index eb0f3dcd3ef409..532e2720d4b094 100644
--- a/packages/mui-codemod/src/deprecations/badge-props/test-cases/expected.js
+++ b/packages/mui-codemod/src/deprecations/badge-props/test-cases/expected.js
@@ -1,44 +1,32 @@
 import { Badge } from '@mui/material';
 
-<Badge
-  slots={{
-    root: ComponentsRoot,
-  }}
-  slotProps={{ root: componentsRootProps }}
-/>;
+<Badge slots={{
+  root: ComponentsRoot
+}} slotProps={{ root: componentsRootProps }} />;
 
 <Badge
   slots={{
     badge: SlotsBadge,
-    root: ComponentsRoot,
+    root: ComponentsRoot
   }}
   slotProps={{
     badge: slotsBadgeProps,
-    root: componentsRootProps,
-  }}
-/>;
+    root: componentsRootProps
+  }} />;
 
 <Badge
   slots={{ root: SlotsRoot, badge: SlotsBadge }}
-  slotProps={{
-    badge: slotsBadgeProps,
-    root: {
-      ...componentsRootProps,
-      ...slotsRootProps,
-    },
-  }}
-/>;
+  slotProps={{ badge: slotsBadgeProps, root: {
+    ...componentsRootProps,
+    ...slotsRootProps
+  } }} />;
 
 <Badge
   slots={{ root: SlotsRoot, badge: SlotsBadge }}
-  slotProps={{
-    root: {
-      ...componentsRootProps,
-      ...slotsRootProps,
-    },
-    badge: {
-      ...componentsBadgeProps,
-      ...slotsBadgeProps,
-    },
-  }}
-/>;
+  slotProps={{ root: {
+    ...componentsRootProps,
+    ...slotsRootProps
+  }, badge: {
+    ...componentsBadgeProps,
+    ...slotsBadgeProps
+  } }} />;
diff --git a/packages/mui-codemod/src/deprecations/badge-props/test-cases/theme.actual.js b/packages/mui-codemod/src/deprecations/badge-props/test-cases/theme.actual.js
index 7743c85238f269..83199128c2c90c 100644
--- a/packages/mui-codemod/src/deprecations/badge-props/test-cases/theme.actual.js
+++ b/packages/mui-codemod/src/deprecations/badge-props/test-cases/theme.actual.js
@@ -28,3 +28,14 @@ fn({
     },
   },
 });
+
+fn({
+  MuiBadge: {
+    defaultProps: {
+      components: { root: ComponentsRoot },
+      slots: { badge: SlotsBadge, root: SlotsRoot },
+      componentsProps: { root: componentsRootProps, badge: componentsBadgeProps },
+      slotProps: { badge: slotsBadgeProps, root: slotsRootProps },
+    },
+  },
+});
diff --git a/packages/mui-codemod/src/deprecations/badge-props/test-cases/theme.expected.js b/packages/mui-codemod/src/deprecations/badge-props/test-cases/theme.expected.js
index f36a27ad59db84..63a3a7f6b893d6 100644
--- a/packages/mui-codemod/src/deprecations/badge-props/test-cases/theme.expected.js
+++ b/packages/mui-codemod/src/deprecations/badge-props/test-cases/theme.expected.js
@@ -2,12 +2,12 @@ fn({
   MuiBadge: {
     defaultProps: {
       slots: {
-        root: ComponentsRoot,
+        root: ComponentsRoot
       },
 
       slotProps: {
-        root: componentsRootProps,
-      },
+        root: componentsRootProps
+      }
     },
   },
 });
@@ -17,13 +17,13 @@ fn({
     defaultProps: {
       slots: {
         root: ComponentsRoot,
-        badge: SlotsBadge,
+        badge: SlotsBadge
       },
 
       slotProps: {
         root: componentsRootProps,
-        badge: slotsBadgeProps,
-      },
+        badge: slotsBadgeProps
+      }
     },
   },
 });
@@ -33,17 +33,40 @@ fn({
     defaultProps: {
       slots: {
         root: SlotsRoot,
-        badge: SlotsBadge,
+        badge: SlotsBadge
       },
 
       slotProps: {
         root: {
           ...componentsRootProps,
-          ...slotsRootProps,
+          ...slotsRootProps
         },
 
-        badge: slotsBadgeProps,
+        badge: slotsBadgeProps
+      }
+    },
+  },
+});
+
+fn({
+  MuiBadge: {
+    defaultProps: {
+      slots: {
+        root: SlotsRoot,
+        badge: SlotsBadge
       },
+
+      slotProps: {
+        root: {
+          ...componentsRootProps,
+          ...slotsRootProps
+        },
+
+        badge: {
+          ...componentsBadgeProps,
+          ...slotsBadgeProps
+        }
+      }
     },
   },
 });
diff --git a/packages/mui-codemod/src/deprecations/slider-props/test-cases/theme.expected.js b/packages/mui-codemod/src/deprecations/slider-props/test-cases/theme.expected.js
index 0ef10d93f7a880..63a3a7f6b893d6 100644
--- a/packages/mui-codemod/src/deprecations/slider-props/test-cases/theme.expected.js
+++ b/packages/mui-codemod/src/deprecations/slider-props/test-cases/theme.expected.js
@@ -1,70 +1,70 @@
 fn({
-  MuiSlider: {
+  MuiBadge: {
     defaultProps: {
       slots: {
-        track: ComponentsTrack
+        root: ComponentsRoot
       },
 
       slotProps: {
-        track: componentsTrackProps
+        root: componentsRootProps
       }
     },
   },
 });
 
 fn({
-  MuiSlider: {
+  MuiBadge: {
     defaultProps: {
       slots: {
-        track: ComponentsTrack,
-        rail: SlotsRail
+        root: ComponentsRoot,
+        badge: SlotsBadge
       },
 
       slotProps: {
-        track: componentsTrackProps,
-        rail: slotsRailProps
+        root: componentsRootProps,
+        badge: slotsBadgeProps
       }
     },
   },
 });
 
 fn({
-  MuiSlider: {
+  MuiBadge: {
     defaultProps: {
       slots: {
-        track: SlotsTrack,
-        rail: SlotsRail
+        root: SlotsRoot,
+        badge: SlotsBadge
       },
 
       slotProps: {
-        track: {
-          ...componentsTrackProps,
-          ...slotsTrackProps
+        root: {
+          ...componentsRootProps,
+          ...slotsRootProps
         },
 
-        rail: slotsRailProps
+        badge: slotsBadgeProps
       }
     },
   },
 });
 
 fn({
-  MuiSlider: {
+  MuiBadge: {
     defaultProps: {
       slots: {
-        track: SlotsTrack,
-        rail: SlotsRail
+        root: SlotsRoot,
+        badge: SlotsBadge
       },
 
       slotProps: {
-        track: {
-          ...componentsTrackProps,
-          ...slotsTrackProps
+        root: {
+          ...componentsRootProps,
+          ...slotsRootProps
         },
 
-        rail: {
-          ...componentsRailProps,
-          ...slotsRailProps
+        badge: {
+          ...componentsBadgeProps,
+          ...slotsBadgeProps
         }
       }
     },