diff --git a/src/AppHeader.tsx b/src/AppHeader.tsx
index 5342ffc..a6de0b1 100644
--- a/src/AppHeader.tsx
+++ b/src/AppHeader.tsx
@@ -4,21 +4,23 @@ import { call_rpc } from "@zmkfirmware/zmk-studio-ts-client";
 import { useConnectedDeviceData } from "./rpc/useConnectedDeviceData";
 import { useSub } from "./usePubSub";
 import { ConnectionContext } from "./rpc/ConnectionContext";
+import { GetDeviceInfoResponse } from "@zmkfirmware/zmk-studio-ts-client/core";
 
 export interface AppHeaderProps {
   connectedDeviceLabel?: string;
 }
 
-export const AppHeader = ({ connectedDeviceLabel }: AppHeaderProps) => {
+export const AppHeader = ({}: AppHeaderProps) => {
   const [unsaved, setUnsaved] = useConnectedDeviceData<boolean>(
     { keymap: { checkUnsavedChanges: true } },
     (r) => r.keymap?.checkUnsavedChanges
   );
 
-  const [deviceInfo, _setDeviceInfo] = useConnectedDeviceData<boolean>(
-    { core: { getDeviceInfo: true } },
-    (r) => r.core?.getDeviceInfo
-  );
+  const [deviceInfo, _setDeviceInfo] =
+    useConnectedDeviceData<GetDeviceInfoResponse>(
+      { core: { getDeviceInfo: true } },
+      (r) => r.core?.getDeviceInfo
+    );
   const conn = useContext(ConnectionContext);
 
   useSub("rpc_notification.keymap.unsavedChangesStatusChanged", (unsaved) =>