diff --git a/site/en/guide/intro_to_modules.ipynb b/site/en/guide/intro_to_modules.ipynb
index 79bbe89ca5..5ea08639f3 100644
--- a/site/en/guide/intro_to_modules.ipynb
+++ b/site/en/guide/intro_to_modules.ipynb
@@ -775,7 +775,7 @@
       },
       "outputs": [],
       "source": [
-        "simple_layer([[2.0, 2.0, 2.0]])"
+        "simple_layer(tf.Variable([[2.0, 2.0, 2.0]]))"
       ]
     },
     {