diff --git a/packages/website/src/features/Deploy/QueueFromGitOpsPage.tsx b/packages/website/src/features/Deploy/QueueFromGitOpsPage.tsx
index d6c1fb9a9..deb0e2872 100644
--- a/packages/website/src/features/Deploy/QueueFromGitOpsPage.tsx
+++ b/packages/website/src/features/Deploy/QueueFromGitOpsPage.tsx
@@ -42,6 +42,7 @@ import {
   useToast,
   VStack,
 } from '@chakra-ui/react';
+import { useConnectModal } from '@rainbow-me/rainbowkit';
 import { ChainBuilderContext, PackageReference } from '@usecannon/builder';
 import _ from 'lodash';
 import NextLink from 'next/link';
@@ -56,13 +57,15 @@ import {
   TransactionRequestBase,
   zeroAddress,
 } from 'viem';
-import { useWriteContract, useAccount, useSwitchChain } from 'wagmi';
+import { useAccount, useSwitchChain, useWriteContract } from 'wagmi';
 import pkg from '../../../package.json';
 import NoncePicker from './NoncePicker';
 import { TransactionDisplay } from './TransactionDisplay';
-import { useConnectModal } from '@rainbow-me/rainbowkit';
 import 'react-diff-view/style/index.css';
 
+// Needed to preapre mock run step with registerAction
+import '@/lib/builder';
+
 export default function QueueFromGitOpsPage() {
   return <QueueFromGitOps />;
 }
diff --git a/packages/website/src/features/Deploy/TransactionDetailsPage.tsx b/packages/website/src/features/Deploy/TransactionDetailsPage.tsx
index 8407a6cc7..8a66e1e0a 100644
--- a/packages/website/src/features/Deploy/TransactionDetailsPage.tsx
+++ b/packages/website/src/features/Deploy/TransactionDetailsPage.tsx
@@ -68,6 +68,9 @@ interface Props {
   sigHash: string;
 }
 
+// Needed to preapre mock run step with registerAction
+import '@/lib/builder';
+
 function TransactionDetailsPage({
   safeAddress,
   chainId,