diff --git a/libs/ui-toolkit/src/components/trading-checkbox/checkbox.spec.tsx b/libs/ui-toolkit/src/components/trading-checkbox/checkbox.spec.tsx
index a0db29d4d..eab21b5f0 100644
--- a/libs/ui-toolkit/src/components/trading-checkbox/checkbox.spec.tsx
+++ b/libs/ui-toolkit/src/components/trading-checkbox/checkbox.spec.tsx
@@ -9,12 +9,12 @@ describe('Checkbox', () => {
 
   it('should render a checked checkbox if specified in state', () => {
     render(<TradingCheckbox label="label" checked={true} />);
-    expect(screen.getByRole('img')).toBeInTheDocument();
+    expect(screen.getByTestId(/icon-/)).toBeInTheDocument();
   });
 
   it('should render an unchecked checkbox if specified in state', () => {
     render(<TradingCheckbox label="unchecked" checked={false} />);
-    expect(screen.queryByRole('img')).not.toBeInTheDocument();
+    expect(screen.queryByTestId(/icon-/)).not.toBeInTheDocument();
   });
 
   it('should render an indeterminate checkbox if specified in state', () => {