diff --git a/lib/Dialect/Torch/Transforms/RefineTypes.cpp b/lib/Dialect/Torch/Transforms/RefineTypes.cpp
index 33293273d..f95ea6e7f 100644
--- a/lib/Dialect/Torch/Transforms/RefineTypes.cpp
+++ b/lib/Dialect/Torch/Transforms/RefineTypes.cpp
@@ -859,7 +859,7 @@ void TypeAnalysis::visitOperation(Operation *op,
     result2Knowledge.dtype = self.dtype;
     incorporateKnowledge(op->getResult(0), result0Knowledge);
     incorporateKnowledge(op->getResult(1), result1Knowledge);
-    incorporateKnowledge(op->getResult(2), result1Knowledge);
+    incorporateKnowledge(op->getResult(2), result2Knowledge);
     return;
   }