diff --git a/quiver_engine/layer_result_generators.py b/quiver_engine/layer_result_generators.py
index ceffabe..2e8446b 100644
--- a/quiver_engine/layer_result_generators.py
+++ b/quiver_engine/layer_result_generators.py
@@ -3,6 +3,6 @@
 
 def get_outputs_generator(model, layer_name):
     return Model(
-        input=model.input,
-        output=model.get_layer(layer_name).output
+        inputs=model.input,
+        outputs=model.get_layer(layer_name).output
     ).predict