diff --git a/docs/examples/DataSet/Working with snapshots.ipynb b/docs/examples/DataSet/Working with snapshots.ipynb
index e918d3696dd..6ac653d0600 100644
--- a/docs/examples/DataSet/Working with snapshots.ipynb	
+++ b/docs/examples/DataSet/Working with snapshots.ipynb	
@@ -678,7 +678,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "assert {\"station\": snapshot_of_station} == snapshot_of_run"
+    "assert {\"station\": snapshot_of_station} == snapshot_of_run[\"station\"]"
    ]
   },
   {