diff --git a/packages/mui-codemod/src/v5.0.0/joy-rename-components-to-slots.test/actual.js b/packages/mui-codemod/src/v5.0.0/joy-rename-components-to-slots.test/actual.js
index 0156a966be03bf..25f77a6d872e0f 100644
--- a/packages/mui-codemod/src/v5.0.0/joy-rename-components-to-slots.test/actual.js
+++ b/packages/mui-codemod/src/v5.0.0/joy-rename-components-to-slots.test/actual.js
@@ -1,4 +1,18 @@
-<Autocomplete
-  components={{ Root, Input: CustomInput }}
-  componentsProps={{ root: { className: 'root' }, input: { 'data-testid': 'input' } }}
-/>;
\ No newline at end of file
+// the codemod should transform only Joy UI components;
+import { Alert as JoyAlert } from '@mui/joy';
+import JoyAutocomplete from '@mui/joy/Autocomplete';
+import CustomComponent from 'components/Custom';
+
+<div>
+  <JoyAlert
+    components={{ Root, Input: CustomInput }}
+    componentsProps={{ root: { className: 'root' }, input: { 'data-testid': 'input' } }}
+  />
+  <JoyAutocomplete
+    components={{ Root, Input: CustomInput }}
+    componentsProps={{ root: { className: 'root' }, input: { 'data-testid': 'input' } }}
+  />
+  <CustomComponent
+    componentsProps={{ root: { className: 'root' }, input: { 'data-testid': 'input' } }}
+  />
+</div>;
diff --git a/packages/mui-codemod/src/v5.0.0/joy-rename-components-to-slots.test/expected.js b/packages/mui-codemod/src/v5.0.0/joy-rename-components-to-slots.test/expected.js
index 8e31e325123497..a46c8938a788e2 100644
--- a/packages/mui-codemod/src/v5.0.0/joy-rename-components-to-slots.test/expected.js
+++ b/packages/mui-codemod/src/v5.0.0/joy-rename-components-to-slots.test/expected.js
@@ -1,4 +1,18 @@
-<Autocomplete
-  slots={{ root: Root, input: CustomInput }}
-  slotProps={{ root: { className: 'root' }, input: { 'data-testid': 'input' } }}
-/>;
\ No newline at end of file
+// the codemod should transform only Joy UI slots;
+import { Alert as JoyAlert } from '@mui/joy';
+import JoyAutocomplete from '@mui/joy/Autocomplete';
+import CustomComponent from 'components/Custom';
+
+<div>
+  <JoyAlert
+    slots={{ Root, Input: CustomInput }}
+    slotProps={{ root: { className: 'root' }, input: { 'data-testid': 'input' } }}
+  />
+  <JoyAutocomplete
+    slots={{ Root, Input: CustomInput }}
+    slotProps={{ root: { className: 'root' }, input: { 'data-testid': 'input' } }}
+  />
+  <CustomComponent
+    slotProps={{ root: { className: 'root' }, input: { 'data-testid': 'input' } }}
+  />
+</div>;