diff --git a/custom_components/extron/extron.py b/custom_components/extron/extron.py
index 3ac2e7a..4e6e323 100644
--- a/custom_components/extron/extron.py
+++ b/custom_components/extron/extron.py
@@ -113,8 +113,8 @@ def __init__(self, device: ExtronDevice) -> None:
     def get_device(self) -> ExtronDevice:
         return self._device
 
-    async def view_input(self):
-        return await self._device.run_command("$")
+    async def view_input(self) -> int:
+        return int(await self._device.run_command("$"))
 
     async def select_input(self, input: int):
         await self._device.run_command(f"{str(input)}$")
@@ -154,8 +154,8 @@ def __init__(self, device: ExtronDevice) -> None:
     def get_device(self) -> ExtronDevice:
         return self._device
 
-    async def view_input(self):
-        return await self._device.run_command("!")
+    async def view_input(self) -> int:
+        return int(await self._device.run_command("!"))
 
     async def select_input(self, input: int):
         await self._device.run_command(f"{str(input)}!")