diff --git a/src/finn/core/onnx_exec.py b/src/finn/core/onnx_exec.py
index 7c3123cd5eb29a54dc5cbfb912225ad3fdb0f219..cee9b703a8c85a405551bf8ea219f168a087dd5a 100644
--- a/src/finn/core/onnx_exec.py
+++ b/src/finn/core/onnx_exec.py
@@ -51,7 +51,8 @@ def execute_node(node, context, graph):
     if node.op_type == "StreamingDataflowPartition":
         sdp_node = getCustomOp(node)
         model = ModelWrapper(sdp_node.get_nodeattr("model"))
-        ret = execute_onnx(model, context, True)
+        inp_ctx = dict(filter(lambda x: x[0] in node.input, context.items()))
+        ret = execute_onnx(model, inp_ctx, False)
         context.update(ret)
     else:
         if node.domain == "finn":