From ac23134e0c82fdc1e073a977012f8784f8d4d075 Mon Sep 17 00:00:00 2001
From: Matthew Corner <matt.corner@ipg-automotive.com>
Date: Thu, 22 Feb 2024 12:04:26 +0000
Subject: [PATCH] Fix types

---
 .../LoadErrorMessage.stories.tsx              | 20 +++++++++++++------
 1 file changed, 14 insertions(+), 6 deletions(-)

diff --git a/src/LoadErrorMessage/LoadErrorMessage.stories.tsx b/src/LoadErrorMessage/LoadErrorMessage.stories.tsx
index c26b2714..465eb42a 100644
--- a/src/LoadErrorMessage/LoadErrorMessage.stories.tsx
+++ b/src/LoadErrorMessage/LoadErrorMessage.stories.tsx
@@ -1,4 +1,4 @@
-import { Meta, StoryFn } from "@storybook/react";
+import { Meta, StoryFn, StoryObj } from "@storybook/react";
 
 import LoadErrorMessage from "./LoadErrorMessage";
 import { LoadErrorMessageProps } from "./LoadErrorMessage.types";
@@ -8,6 +8,11 @@ import React from "react";
  * Story metadata
  */
 const meta: Meta<typeof LoadErrorMessage> = {
+  argTypes: {
+    onButtonClick: {
+      control: false
+    }
+  },
   component: LoadErrorMessage,
   title: "Card/LoadErrorMessage"
 };
@@ -18,8 +23,11 @@ const Template: StoryFn<LoadErrorMessageProps> = args => {
   return <LoadErrorMessage {...args} />;
 };
 
+// Story type
+type Story = StoryObj<typeof LoadErrorMessage>;
+
 // Default
-export const Default = {
+export const Default: Story = {
   args: {
     contactTeam: "Support",
     image: "virto-thinking",
@@ -30,7 +38,7 @@ export const Default = {
 };
 
 // HTTP 404
-export const HTTP404 = {
+export const HTTP404: Story = {
   args: {
     ...Default.args,
     actionButtonText: "Go back",
@@ -44,7 +52,7 @@ export const HTTP404 = {
 };
 
 // HTTP 401
-export const HTTP401 = {
+export const HTTP401: Story = {
   args: {
     ...Default.args,
     actionButtonText: undefined,
@@ -57,15 +65,15 @@ export const HTTP401 = {
 };
 
 // General Error
-export const GeneralError = {
+export const GeneralError: Story = {
   args: {
     ...Default.args,
     actionButtonText: "Refresh",
+    contactUrl: "https://support.virto.com",
     errorDetails: "Invalid token - length is 0",
     image: "virto-head-scratching",
     message:
       "Oops! Something went wrong on our end 😓 Our team is actively addressing it and working to resolve the issue. Please try again later.",
-    contactUrl: "https://support.virto.com",
     title: "Something is not right!"
   },
   render: Template