diff --git a/tutorials/OnnxTensorflowImport.ipynb b/tutorials/OnnxTensorflowImport.ipynb
index 433ce2a..e066d7c 100644
--- a/tutorials/OnnxTensorflowImport.ipynb
+++ b/tutorials/OnnxTensorflowImport.ipynb
@@ -39,7 +39,7 @@
     "\n",
     "warnings.filterwarnings('ignore') # Ignore all the warning messages in this tutorial\n",
     "model = onnx.load('assets/super_resolution.onnx') # Load the ONNX file\n",
-    "tf_rep = prepare(model) # Import the ONNX model to Tensorflow"
+    "tf_rep = prepare(model, gen_tensor_dict=True) # Import the ONNX model to Tensorflow"
    ]
   },
   {