diff --git a/zksnark/bn128_curve.py b/zksnark/bn128_curve.py
index 368a84d..64bcfc0 100644
--- a/zksnark/bn128_curve.py
+++ b/zksnark/bn128_curve.py
@@ -1,4 +1,5 @@
-from bn128_field_elements import field_modulus, FQ, FQ2, FQ12
+from bn128_field_elements import field_modulus, FQ
+from optimized_field_elements import FQ2, FQ12
 
 curve_order = 21888242871839275222246405745257275088548364400416034343698204186575808495617