diff --git a/mlir/test/Integration/Dialect/SparseTensor/taco/test_Tensor.py b/mlir/test/Integration/Dialect/SparseTensor/taco/test_Tensor.py
index 08710cd429c8..5b8b687562de 100644
--- a/mlir/test/Integration/Dialect/SparseTensor/taco/test_Tensor.py
+++ b/mlir/test/Integration/Dialect/SparseTensor/taco/test_Tensor.py
@@ -25,17 +25,15 @@ S.insert([1, 1, 1], 3.0)
 S.insert([4, 4, 4], 4.0)
 S.insert([7, 7, 7], 5.0)
 
-# TODO: make this work:
-# X[i, j, k] = alpha[0] * S[i, j, k]
-X[i, j, k] = S[i, j, k]
+X[i, j, k] = alpha[0] * S[i, j, k]
 
 expected = """; extended FROSTT format
 3 4
 8 8 8
-1 1 1 2
-2 2 2 3
-5 5 5 4
-8 8 8 5
+1 1 1 84
+2 2 2 126
+5 5 5 168
+8 8 8 210
 """
 
 # Force evaluation of the kernel by writing out X.