diff --git a/main.py b/main.py
index 5b9b317..01e8079 100644
--- a/main.py
+++ b/main.py
@@ -21,6 +21,11 @@
     is_inf
 )
 
+from py_ecc.optimized_bls12_381 import (
+    FQ2,
+    neg
+)
+
 from py_ecc.bls.hash_to_curve import hash_to_G2
 
 def to_bytes32(i):