small change
fixed bug with multiple outputs