diff --git a/main.py b/main.py
index d5d9329..6be60a4 100644
--- a/main.py
+++ b/main.py
@@ -676,7 +676,7 @@ if __name__ == "__main__":
                 neg_ent = neg_ent.cpu()
 
             inputs = []
-            for i in len(sub):
+            for i in range(len(sub)):
                 input = {}
                 input['sub'] = sub[i]
                 input['rel'] = rel[i]