diff --git a/plugins/catalog-graph/src/components/EntityRelationsGraph/EntityKindIcon.test.tsx b/plugins/catalog-graph/src/components/EntityRelationsGraph/EntityKindIcon.test.tsx
index 6592c6195e..cc47c8850b 100644
--- a/plugins/catalog-graph/src/components/EntityRelationsGraph/EntityKindIcon.test.tsx
+++ b/plugins/catalog-graph/src/components/EntityRelationsGraph/EntityKindIcon.test.tsx
@@ -24,7 +24,7 @@ describe('<EntityKindIcon />', () => {
       <EntityKindIcon kind="Component" />,
     );
 
-    expect(baseElement.querySelector('.MuiSvgIcon-root')).toBeInTheDocument();
+    expect(baseElement.querySelector('svg')).toBeInTheDocument();
   });
 
   it('renders without exploding for unknown kind', async () => {
@@ -32,6 +32,6 @@ describe('<EntityKindIcon />', () => {
       <EntityKindIcon kind="unknown" />,
     );
 
-    expect(baseElement.querySelector('.MuiSvgIcon-root')).toBeInTheDocument();
+    expect(baseElement.querySelector('svg')).toBeInTheDocument();
   });
 });