diff --git a/blockchain/diophantine_equation.py b/blockchain/diophantine_equation.py
index ec2ed26e4..dab4b3a65 100644
--- a/blockchain/diophantine_equation.py
+++ b/blockchain/diophantine_equation.py
@@ -118,10 +118,9 @@ def extended_gcd(a, b):
     return (d, x, y)
 
 
-# import testmod for testing our function
-from doctest import testmod
-
 if __name__ == "__main__":
+    from doctest import testmod
+
     testmod(name="diophantine", verbose=True)
     testmod(name="diophantine_all_soln", verbose=True)
     testmod(name="extended_gcd", verbose=True)
diff --git a/blockchain/modular_division.py b/blockchain/modular_division.py
index 1255f0432..c81c2138d 100644
--- a/blockchain/modular_division.py
+++ b/blockchain/modular_division.py
@@ -139,10 +139,9 @@ def greatest_common_divisor(a, b):
     return b
 
 
-# Import testmod for testing our function
-from doctest import testmod
-
 if __name__ == "__main__":
+    from doctest import testmod
+
     testmod(name="modular_division", verbose=True)
     testmod(name="modular_division2", verbose=True)
     testmod(name="invert_modulo", verbose=True)