diff --git a/src/__mocks__/shared-mocks.tsx b/src/__mocks__/shared-mocks.tsx
index fe6df15..3f5d063 100644
--- a/src/__mocks__/shared-mocks.tsx
+++ b/src/__mocks__/shared-mocks.tsx
@@ -89,6 +89,7 @@ export const mockLucideReact = () => {
     Zap: () => <span data-testid="zap-icon" />,
     Award: () => <span data-testid="award-icon" />,
     Check: () => <span data-testid="check-icon" />,
+    CheckCircle2: () => <span data-testid="check-circle-icon" />,
     X: () => <span data-testid="x-icon" />,
     Menu: () => <span data-testid="menu-icon" />,
     ChevronDown: () => <span data-testid="chevron-down" />,
@@ -97,6 +98,8 @@ export const mockLucideReact = () => {
     Phone: () => <span data-testid="phone-icon" />,
     MapPin: () => <span data-testid="map-pin-icon" />,
     Clock: () => <span data-testid="clock-icon" />,
+    Calendar: () => <span data-testid="calendar-icon" />,
+    Quote: () => <span data-testid="quote-icon" />,
     User: () => <span data-testid="user-icon" />,
     Lock: () => <span data-testid="lock-icon" />,
     Eye: () => <span data-testid="eye-icon" />,
diff --git a/src/components/ui/toast.test.tsx b/src/components/ui/toast.test.tsx
index 218715e..1fe49aa 100644
--- a/src/components/ui/toast.test.tsx
+++ b/src/components/ui/toast.test.tsx
@@ -125,19 +125,19 @@ describe('Toast Component', () => {
     it('should render success icon for success type', () => {
       render(<Toast message="Success" type="success" onClose={mockOnClose} />);
       const toast = screen.getByRole('alert');
-      expect(toast.querySelector('svg')).toBeInTheDocument();
+      expect(toast.querySelector('[data-testid]')).toBeInTheDocument();
     });
 
     it('should render error icon for error type', () => {
       render(<Toast message="Error" type="error" onClose={mockOnClose} />);
       const toast = screen.getByRole('alert');
-      expect(toast.querySelector('svg')).toBeInTheDocument();
+      expect(toast.querySelector('[data-testid]')).toBeInTheDocument();
     });
 
     it('should render info icon for info type', () => {
       render(<Toast message="Info" type="info" onClose={mockOnClose} />);
       const toast = screen.getByRole('alert');
-      expect(toast.querySelector('svg')).toBeInTheDocument();
+      expect(toast.querySelector('[data-testid]')).toBeInTheDocument();
     });
   });