diff --git a/site/src/modules/resources/PortForwardButton.stories.tsx b/site/src/modules/resources/PortForwardButton.stories.tsx
index 6d09f321c4cd6..ddf683d37b078 100644
--- a/site/src/modules/resources/PortForwardButton.stories.tsx
+++ b/site/src/modules/resources/PortForwardButton.stories.tsx
@@ -6,10 +6,12 @@ import {
   MockWorkspace,
   MockWorkspaceAgent,
 } from "testHelpers/entities";
+import { withDashboardProvider } from "testHelpers/storybook";
 
 const meta: Meta<typeof PortForwardButton> = {
   title: "modules/resources/PortForwardButton",
   component: PortForwardButton,
+  decorators: [withDashboardProvider],
   args: {
     agent: MockWorkspaceAgent,
   },