diff --git a/python/tasks/jigsaw_toxic.py b/python/tasks/jigsaw_toxic.py
index 388839b..b0383e9 100644
--- a/python/tasks/jigsaw_toxic.py
+++ b/python/tasks/jigsaw_toxic.py
@@ -295,4 +295,4 @@ def kernel_3(
     #   Try printing first 2 elements of xtrain_seq you will see every word is represented as a digit now
 
     # %% [code]
-    xtrain_seq[:1]
+    o_2['xtrain_seq'][:1]