diff --git a/hsf/engines.py b/hsf/engines.py
index 8635b0f..809ab49 100644
--- a/hsf/engines.py
+++ b/hsf/engines.py
@@ -103,7 +103,9 @@ def __init__(self, engine_name: str, engine_settings: DictConfig,
 
     def __call__(self, x):
         if self.engine_name == "onnxruntime":
-            return self.engine.run(None, {"input": x})
+            feed_names = [i.name for i in self.engine.get_inputs()]
+            assert len(feed_names) == 1, "Only one input is supported"
+            return self.engine.run(None, {feed_names[0]: x})
 
         elif self.engine_name == "deepsparse":
             return self.engine.run([x])