1 files changed, 3 insertions, 1 deletions

diff --git a/test/ec/perf_mod.py b/test/ec/perf_mod.py
index 925360e..8f07fc5 100755
--- a/test/ec/perf_mod.py
+++ b/test/ec/perf_mod.py
@@ -1,7 +1,9 @@
 #!/usr/bin/env python
 import click
 
-from pyecsca.ec.mod import Mod, has_gmp, has_flint
+from pyecsca.ec.mod import Mod
+from pyecsca.ec.mod.gmp import has_gmp
+from pyecsca.ec.mod.flint import has_flint
 from pyecsca.misc.cfg import TemporaryConfig
 from test.utils import Profiler