diff --git a/apps/web/src/lib/scientific/useHeatmap.test.tsx b/apps/web/src/lib/scientific/useHeatmap.test.tsx
index 3218381..760c411 100644
--- a/apps/web/src/lib/scientific/useHeatmap.test.tsx
+++ b/apps/web/src/lib/scientific/useHeatmap.test.tsx
@@ -110,8 +110,10 @@ describe('useHeatmap', () => {
     const loader = vi.fn().mockResolvedValueOnce(first).mockResolvedValueOnce(second)
     const captured = renderHook({ loader })
     await waitFor(() => expect(screen.getByTestId('status').textContent).toBe('success'))
-    act(() => captured.model.selectCell(heatmapCellKey({ row: 'r1', column: 'c1' })))
-    await waitFor(() => expect(captured.model.selectedKey).not.toBeNull())
+    await act(async () => {})
+    const key = heatmapCellKey({ row: 'r1', column: 'c1' })
+    act(() => captured.model.selectCell(key))
+    await waitFor(() => expect(captured.model.selectedKey).toBe(key))
     act(() => captured.model.refetch())
     await waitFor(() => expect(captured.model.dataset?.id).toBe('b'))
     await waitFor(() => expect(captured.model.selectedKey).toBeNull())