diff --git a/src/iTPS/multisite_obs.cpp b/src/iTPS/multisite_obs.cpp
index 498ad014..f337066c 100644
--- a/src/iTPS/multisite_obs.cpp
+++ b/src/iTPS/multisite_obs.cpp
@@ -156,6 +156,7 @@ auto iTPS<ptensor>::measure_multisite()
           indices[row][col] = index;
           op_[row][col] = &(op_identity[index]);
           Tn_[row][col] = &(Tn[index]);
+          shape_types[row * ncol + col] = tensor_shape_types[index];
         }
         eTl_[row] = &(eTl[indices[row][0]]);
         eTr_[row] = &(eTr[indices[row][ncol - 1]]);
diff --git a/src/iTPS/twosite_obs.cpp b/src/iTPS/twosite_obs.cpp
index ca7146de..7580dc64 100644
--- a/src/iTPS/twosite_obs.cpp
+++ b/src/iTPS/twosite_obs.cpp
@@ -158,6 +158,7 @@ auto iTPS<ptensor>::measure_twosite()
           indices[row][col] = index;
           op_[row][col] = &(op_identity[index]);
           Tn_[row][col] = &(Tn[index]);
+          shape_types[row * ncol + col] = tensor_shape_types[index];
         }
         eTl_[row] = &(eTl[indices[row][0]]);
         eTr_[row] = &(eTr[indices[row][ncol - 1]]);