diff --git a/src/frontend/src/components/codeTabsComponent/index.tsx b/src/frontend/src/components/codeTabsComponent/index.tsx
index 9868b709a..67ad34a7e 100644
--- a/src/frontend/src/components/codeTabsComponent/index.tsx
+++ b/src/frontend/src/components/codeTabsComponent/index.tsx
@@ -56,7 +56,7 @@ export default function CodeTabsComponent({
   }, [flow]);
 
   useEffect(() => {
-    if (tweaks) {
+    if (tweaks && data) {
       unselectAllNodes({
         data,
         updateNodes: (nodes) => {
diff --git a/src/frontend/src/modals/codeAreaModal/index.tsx b/src/frontend/src/modals/codeAreaModal/index.tsx
index 0dc6a2a9e..5851017ef 100644
--- a/src/frontend/src/modals/codeAreaModal/index.tsx
+++ b/src/frontend/src/modals/codeAreaModal/index.tsx
@@ -28,7 +28,7 @@ export default function CodeAreaModal({
   const { dark } = useContext(darkContext);
   const { reactFlowInstance } = useContext(typesContext);
   const [height, setHeight] = useState<string | null>(null);
-  const { setErrorData, setSuccessData, isTweakPage } =
+  const { setErrorData, setSuccessData } =
     useContext(alertContext);
   const [error, setError] = useState<{
     detail: { error: string | undefined; traceback: string | undefined };
diff --git a/src/frontend/src/types/utils/reactflowUtils.ts b/src/frontend/src/types/utils/reactflowUtils.ts
index e9b8ee473..1ce069782 100644
--- a/src/frontend/src/types/utils/reactflowUtils.ts
+++ b/src/frontend/src/types/utils/reactflowUtils.ts
@@ -12,7 +12,7 @@ export type cleanEdgesType = {
 
 export type unselectAllNodesType = {
   updateNodes: (nodes: Node[]) => void;
-  data: Node[] | null;
+  data: Node[];
 };
 
 export type updateEdgesHandleIdsType = {
diff --git a/src/frontend/src/utils/reactflowUtils.ts b/src/frontend/src/utils/reactflowUtils.ts
index 89df35130..7818f3965 100644
--- a/src/frontend/src/utils/reactflowUtils.ts
+++ b/src/frontend/src/utils/reactflowUtils.ts
@@ -75,7 +75,7 @@ export function cleanEdges({
 
 export function unselectAllNodes({ updateNodes, data }: unselectAllNodesType) {
   let newNodes = _.cloneDeep(data);
-  newNodes!.forEach((node: Node) => {
+  newNodes.forEach((node: Node) => {
     node.selected = false;
   });
   updateNodes(newNodes!);