diff --git a/packages/react/src/Image/Image.test.tsx b/packages/react/src/Image/Image.test.tsx
index 26fd20aaba..6224efb8ed 100644
--- a/packages/react/src/Image/Image.test.tsx
+++ b/packages/react/src/Image/Image.test.tsx
@@ -26,8 +26,7 @@ describe('Image', () => {
 
     const component = container.querySelector(':only-child')
 
-    expect(component).toHaveClass('extra')
-    expect(component).toHaveClass('ams-image')
+    expect(component).toHaveClass('ams-image extra')
   })
 
   it('renders class names to display the image in a given aspect ratio', () => {
diff --git a/packages/react/src/Switch/Switch.test.tsx b/packages/react/src/Switch/Switch.test.tsx
index a21b62e6df..7cef895e7f 100644
--- a/packages/react/src/Switch/Switch.test.tsx
+++ b/packages/react/src/Switch/Switch.test.tsx
@@ -92,7 +92,7 @@ describe('Switch', () => {
 
     const component = container.querySelector(':only-child')
 
-    expect(component).toHaveClass('ams-switch', 'extra')
+    expect(component).toHaveClass('ams-switch extra')
   })
 
   it('is able to pass a React ref', () => {