From 6b838756788103525784c1498d71309f9b90b1bc Mon Sep 17 00:00:00 2001
From: BrunoQuaresma <bruno_nonato_quaresma@hotmail.com>
Date: Wed, 21 Feb 2024 14:27:38 +0000
Subject: [PATCH] chore(site): fix storybook for agent row with port forward
 button

---
 site/src/modules/resources/AgentRow.stories.tsx | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/site/src/modules/resources/AgentRow.stories.tsx b/site/src/modules/resources/AgentRow.stories.tsx
index fea33c7c4acef..c2cf5703446b9 100644
--- a/site/src/modules/resources/AgentRow.stories.tsx
+++ b/site/src/modules/resources/AgentRow.stories.tsx
@@ -24,6 +24,7 @@ import {
 } from "testHelpers/entities";
 import { AgentRow, LineWithID } from "./AgentRow";
 import { ProxyContext, getPreferredProxy } from "contexts/ProxyContext";
+import { withDashboardProvider } from "testHelpers/storybook";
 
 const defaultAgentMetadata = [
   {
@@ -113,6 +114,7 @@ const meta: Meta<typeof AgentRow> = {
       },
     ],
   },
+
   component: AgentRow,
   args: {
     storybookLogs,
@@ -147,6 +149,7 @@ const meta: Meta<typeof AgentRow> = {
         <Story />
       </ProxyContext.Provider>
     ),
+    withDashboardProvider,
   ],
 };