diff --git a/DocumentUnderstanding/VGT/object_detection/ditod/Wordnn_embedding.py b/DocumentUnderstanding/VGT/object_detection/ditod/Wordnn_embedding.py
index 11bf984..0057d45 100644
--- a/DocumentUnderstanding/VGT/object_detection/ditod/Wordnn_embedding.py
+++ b/DocumentUnderstanding/VGT/object_detection/ditod/Wordnn_embedding.py
@@ -77,7 +77,7 @@ def forward(self, img, batched_inputs, stride = 1):
                     per_id = per_input_ids[word_idx]
                     
                     bbox = per_input_bbox[word_idx] / stride
-                    w_start, h_start, w_end, h_end = bbox.round().astype(np.int).tolist()
+                    w_start, h_start, w_end, h_end = bbox.round().astype(int).tolist()
                             
                     if self.use_UNK_text:
                         chargrid_map[iter_b, h_start:h_end, w_start: w_end] = 100