diff --git a/webview-ui/src/App.tsx b/webview-ui/src/App.tsx
index 31547ef..971926b 100644
--- a/webview-ui/src/App.tsx
+++ b/webview-ui/src/App.tsx
@@ -91,6 +91,9 @@ const App: React.FC = () => {
 								setShowHistory(false)
 							}
 							break
+						case "koduCreditsFetched":
+							setKoduCredits(message.state!.koduCredits)
+							break
 					}
 					break
 			}