diff --git a/notebooks/visualize_exported_dataset.ipynb b/notebooks/visualize_exported_dataset.ipynb
index 348fbd3..5ca610d 100644
--- a/notebooks/visualize_exported_dataset.ipynb
+++ b/notebooks/visualize_exported_dataset.ipynb
@@ -48,6 +48,7 @@
     "# Initialize the wireframe dataset\n",
     "with open(\"../sold2/config/wireframe_dataset.yaml\", \"r\") as f:\n",
     "    config = yaml.safe_load(f)\n",
+    "config['return_type'] = 'paired_desc'\n",
     "\n",
     "wireframe_dataset = WireframeDataset(mode=\"test\", config=config)"
    ]
@@ -390,7 +391,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.7.5"
+   "version": "3.6.9"
   },
   "mimetype": "text/x-python",
   "name": "python",