diff --git a/g3doc/tutorials/graph_keras_lstm_imdb.ipynb b/g3doc/tutorials/graph_keras_lstm_imdb.ipynb
index b0018a5..cba06cf 100644
--- a/g3doc/tutorials/graph_keras_lstm_imdb.ipynb
+++ b/g3doc/tutorials/graph_keras_lstm_imdb.ipynb
@@ -645,7 +645,7 @@
       "outputs": [],
       "source": [
         "!python -m neural_structured_learning.tools.pack_nbrs \\\n",
-        "--max_nbrs=1 --add_undirected_edges=True \\\n",
+        "--max_nbrs=3 --add_undirected_edges=True \\\n",
         "/tmp/imdb/train_data.tfr '' /tmp/imdb/graph_99.tsv \\\n",
         "/tmp/imdb/nsl_train_data.tfr"
       ]