modify mock func

Opentrons · Mar 1, 2024 · 1a6d6fd · 1a6d6fd
1 parent 28b626f
commit 1a6d6fd
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/app/src/organisms/CalibrationPanels/__tests__/DeckSetup.test.tsx b/app/src/organisms/CalibrationPanels/__tests__/DeckSetup.test.tsx
@@ -14,12 +14,16 @@ import * as Sessions from '../../../redux/sessions'
 
 import { DeckSetup } from '../DeckSetup'
 
+import type * as OpentronsComponent from '@opentrons/components'
+
 vi.mock('../../../assets/labware/getLabware')
 vi.mock('@opentrons/shared-data')
 vi.mock(
   '@opentrons/components/src/hardware-sim/Deck/RobotWorkSpace',
   async importOriginal => {
-    const actual = (await importOriginal()) as any
+    const actual = await importOriginal<
+      typeof OpentronsComponent.RobotWorkSpace
+    >()
     return {
       ...actual,
       RobotWorkSpace: () => <></>,