diff --git a/sytorch/include/sytorch/tensor.h b/sytorch/include/sytorch/tensor.h
index 87dfd024..c07c2baf 100644
--- a/sytorch/include/sytorch/tensor.h
+++ b/sytorch/include/sytorch/tensor.h
@@ -235,7 +235,7 @@ class Tensor {
             u64 curr_rest = i % rest_size;
             u64 new_idx = curr_batch * (num_channel * rest_size) + curr_rest * num_channel + curr_channel;
 #ifdef Do_Masking
-            data[new_idx] = type_cast<T> d;
+            data[new_idx] = type_cast<T>(d);
 #else
             data[new_idx] = type_cast<T>(d * (1LL << scale));
 #endif