From 612d71a918bf555c26c08f9db0388a507e1f6c16 Mon Sep 17 00:00:00 2001
From: Oliver Walters <oliver.henry.walters@gmail.com>
Date: Thu, 21 Nov 2024 03:11:59 +0000
Subject: [PATCH] Fix lookup for useStatusCodes

---
 src/frontend/src/components/render/StatusRenderer.tsx |  4 +++-
 src/frontend/src/hooks/UseStatusCodes.tsx             | 10 +++++++---
 2 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/src/frontend/src/components/render/StatusRenderer.tsx b/src/frontend/src/components/render/StatusRenderer.tsx
index 16b6ab7f68..843ad574c0 100644
--- a/src/frontend/src/components/render/StatusRenderer.tsx
+++ b/src/frontend/src/components/render/StatusRenderer.tsx
@@ -68,7 +68,9 @@ function renderStatusLabel(
   );
 }
 
-export function getStatusCodes(type: ModelType | string) {
+export function getStatusCodes(
+  type: ModelType | string
+): StatusCodeListInterface | null {
   const statusCodeList = useGlobalStatusState.getState().status;
 
   if (statusCodeList === undefined) {
diff --git a/src/frontend/src/hooks/UseStatusCodes.tsx b/src/frontend/src/hooks/UseStatusCodes.tsx
index fcc29bd060..9316ea9adf 100644
--- a/src/frontend/src/hooks/UseStatusCodes.tsx
+++ b/src/frontend/src/hooks/UseStatusCodes.tsx
@@ -31,14 +31,18 @@ export default function useStatusCodes({
   const statusCodeList = useGlobalStatusState.getState().status;
 
   const codes = useMemo(() => {
-    const statusCodes = getStatusCodes(modelType) || {};
+    const statusCodes = getStatusCodes(modelType) || null;
 
     const codesMap: Record<any, any> = {};
 
-    for (const name in statusCodes) {
-      codesMap[name] = statusCodes[name].key;
+    if (!statusCodes) {
+      return codesMap;
     }
 
+    Object.keys(statusCodes.values).forEach((name) => {
+      codesMap[name] = statusCodes.values[name].key;
+    });
+
     return codesMap;
   }, [modelType, statusCodeList]);