diff --git a/server/vad.py b/server/vad.py
index 15f28f0..7740a85 100644
--- a/server/vad.py
+++ b/server/vad.py
@@ -18,8 +18,7 @@ class SileroVADOnnx:
         self._reset_state()
 
     def _reset_state(self):
-        self._h = np.zeros((2, 1, 64), dtype=np.float32)
-        self._c = np.zeros((2, 1, 64), dtype=np.float32)
+        self._state = np.zeros((2, 1, 128), dtype=np.float32)
 
     def reset_states(self):
         self._reset_state()
@@ -32,10 +31,9 @@ class SileroVADOnnx:
         ort_inputs = {
             "input": input_data,
             "sr": sr,
-            "h": self._h,
-            "c": self._c,
+            "state": self._state,
         }
-        out, self._h, self._c = self.session.run(None, ort_inputs)
+        out, self._state = self.session.run(None, ort_inputs)
         return float(out.squeeze())