diff --git a/mlir/docs/Tutorials/Toy/Ch-1.md b/mlir/docs/Tutorials/Toy/Ch-1.md
--- a/mlir/docs/Tutorials/Toy/Ch-1.md
+++ b/mlir/docs/Tutorials/Toy/Ch-1.md
@@ -61,9 +61,9 @@
   # trigger another specialization of `multiply_transpose`.
   var e = multiply_transpose(c, d);
 
-  # Finally, calling into `multiply_transpose` with incompatible shape will
-  # trigger a shape inference error.
-  var f = multiply_transpose(transpose(a), c);
+  # Finally, calling into `multiply_transpose` with incompatible shapes
+  # (<2, 3> and <3, 2>) will trigger a shape inference error.
+  var f = multiply_transpose(a, c);
 }
 ```
 
@@ -74,7 +74,7 @@
 ```
 Module:
   Function 
-    Proto 'multiply_transpose' @test/Examples/Toy/Ch1/ast.toy:4:1'
+    Proto 'multiply_transpose' @test/Examples/Toy/Ch1/ast.toy:4:1
     Params: [a, b]
     Block {
       Return
@@ -87,7 +87,7 @@
           ]
     } // Block
   Function 
-    Proto 'main' @test/Examples/Toy/Ch1/ast.toy:8:1'
+    Proto 'main' @test/Examples/Toy/Ch1/ast.toy:8:1
     Params: []
     Block {
       VarDecl a<> @test/Examples/Toy/Ch1/ast.toy:11:3
@@ -111,10 +111,8 @@
         ]
       VarDecl f<> @test/Examples/Toy/Ch1/ast.toy:28:3
         Call 'multiply_transpose' [ @test/Examples/Toy/Ch1/ast.toy:28:11
-          Call 'transpose' [ @test/Examples/Toy/Ch1/ast.toy:28:30
-            var: a @test/Examples/Toy/Ch1/ast.toy:28:40
-          ]
-          var: c @test/Examples/Toy/Ch1/ast.toy:28:44
+          var: a @test/Examples/Toy/Ch1/ast.toy:28:30
+          var: c @test/Examples/Toy/Ch1/ast.toy:28:33
         ]
     } // Block
 ```
diff --git a/mlir/test/Examples/Toy/Ch1/ast.toy b/mlir/test/Examples/Toy/Ch1/ast.toy
--- a/mlir/test/Examples/Toy/Ch1/ast.toy
+++ b/mlir/test/Examples/Toy/Ch1/ast.toy
@@ -23,9 +23,9 @@
   # A new call with `<3, 2>` for both dimension will trigger another
   # specialization of `multiply_transpose`.
   var e = multiply_transpose(c, d);
-  # Finally, calling into `multiply_transpose` with incompatible shape will
-  # trigger a shape inference error.
-  var f = multiply_transpose(transpose(a), c);
+  # Finally, calling into `multiply_transpose` with incompatible shapes
+  # (<2, 3> and <3, 2>) will trigger a shape inference error.
+  var f = multiply_transpose(a, c);
 }
 
 
@@ -68,9 +68,7 @@
 # CHECK-NEXT:           ]
 # CHECK-NEXT:         VarDecl f<> @{{.*}}ast.toy:28:3
 # CHECK-NEXT:           Call 'multiply_transpose' [ @{{.*}}ast.toy:28:11
-# CHECK-NEXT:             Call 'transpose' [ @{{.*}}ast.toy:28:30
-# CHECK-NEXT:               var: a @{{.*}}ast.toy:28:40
-# CHECK-NEXT:             ]
-# CHECK-NEXT:             var: c @{{.*}}ast.toy:28:44
+# CHECK-NEXT:             var: a @{{.*}}ast.toy:28:30
+# CHECK-NEXT:             var: c @{{.*}}ast.toy:28:33
 # CHECK-NEXT:           ]
-
+# CHECK-NEXT:       } // Block