diff --git a/Packages/com.github.asus4.tflite.common/Runtime/InterpreterExtension.cs b/Packages/com.github.asus4.tflite.common/Runtime/InterpreterExtension.cs
index b22575f0..18f8be18 100644
--- a/Packages/com.github.asus4.tflite.common/Runtime/InterpreterExtension.cs
+++ b/Packages/com.github.asus4.tflite.common/Runtime/InterpreterExtension.cs
@@ -54,7 +54,7 @@ public static void LogIOInfo(this SignatureRunner runner)
             }
             sb.AppendLine();
 
-            int signatureInputCount = runner.GetSignatureInputCount();
+            int signatureInputCount = (int)runner.GetSignatureInputCount();
             for (int i = 0; i < signatureInputCount; i++)
             {
                 string name = runner.GetSignatureInputName(i);
@@ -62,7 +62,7 @@ public static void LogIOInfo(this SignatureRunner runner)
             }
             sb.AppendLine();
 
-            int signatureOutputCount = runner.GetSignatureOutputCount();
+            int signatureOutputCount = (int)runner.GetSignatureOutputCount();
             for (int i = 0; i < signatureOutputCount; i++)
             {
                 string name = runner.GetSignatureOutputName(i);
diff --git a/Packages/com.github.asus4.tflite/Runtime/SignatureRunner.cs b/Packages/com.github.asus4.tflite/Runtime/SignatureRunner.cs
index e0f07cfb..903cd5c5 100644
--- a/Packages/com.github.asus4.tflite/Runtime/SignatureRunner.cs
+++ b/Packages/com.github.asus4.tflite/Runtime/SignatureRunner.cs
@@ -73,7 +73,7 @@ public string GetSignatureKey(int index)
             return ToString(TfLiteInterpreterGetSignatureKey(InterpreterPointer, index));
         }
 
-        public int GetSignatureInputCount()
+        public ulong GetSignatureInputCount()
         {
             return TfLiteSignatureRunnerGetInputCount(runner);
         }
@@ -138,7 +138,7 @@ public override void Invoke()
             ThrowIfError(TfLiteSignatureRunnerInvoke(runner));
         }
 
-        public int GetSignatureOutputCount()
+        public ulong GetSignatureOutputCount()
         {
             return TfLiteSignatureRunnerGetOutputCount(runner);
         }
@@ -199,7 +199,7 @@ private void Initialize(string signatureName)
 
         private Dictionary<string, int> CreateMap(bool isInput)
         {
-            int signatureCount = isInput ? GetSignatureInputCount() : GetSignatureOutputCount();
+            int signatureCount = (int)(isInput ? GetSignatureInputCount() : GetSignatureOutputCount());
             int tensorCount = isInput ? GetInputTensorCount() : GetOutputTensorCount();
             Assert.AreEqual(signatureCount, tensorCount);
 
@@ -270,7 +270,7 @@ private Dictionary<string, int> CreateMap(bool isInput)
         private static extern TfLiteSignatureRunner TfLiteInterpreterGetSignatureRunner(TfLiteInterpreter interpreter, string signature_name);
 
         [DllImport(TensorFlowLibrary)]
-        private static extern int TfLiteSignatureRunnerGetInputCount(TfLiteSignatureRunner signature_runner);
+        private static extern ulong TfLiteSignatureRunnerGetInputCount(TfLiteSignatureRunner signature_runner);
 
         [DllImport(TensorFlowLibrary)]
         private static extern IntPtr TfLiteSignatureRunnerGetInputName(TfLiteSignatureRunner signature_runner, int input_index);
@@ -290,7 +290,7 @@ private static extern Status TfLiteSignatureRunnerResizeInputTensor(
         private static extern Status TfLiteSignatureRunnerInvoke(TfLiteSignatureRunner signature_runner);
 
         [DllImport(TensorFlowLibrary)]
-        private static extern int TfLiteSignatureRunnerGetOutputCount(TfLiteSignatureRunner signature_runner);
+        private static extern ulong TfLiteSignatureRunnerGetOutputCount(TfLiteSignatureRunner signature_runner);
 
         [DllImport(TensorFlowLibrary)]
         private static extern IntPtr TfLiteSignatureRunnerGetOutputName(TfLiteSignatureRunner signature_runner, int output_index);