diff --git a/tests/test_compare_analytical.py b/tests/test_compare_analytical.py
index 7d829301..1a74facf 100644
--- a/tests/test_compare_analytical.py
+++ b/tests/test_compare_analytical.py
@@ -16,8 +16,9 @@
# You should have received a copy of the GNU General Public License
# along with gprMax. If not, see .
-
-import sys, os, argparse
+import argparse
+import os
+import sys
import h5py
import numpy as np
diff --git a/tests/test_compare_experimental.py b/tests/test_compare_experimental.py
index 961c8b07..33c40c15 100644
--- a/tests/test_compare_experimental.py
+++ b/tests/test_compare_experimental.py
@@ -16,7 +16,10 @@
# You should have received a copy of the GNU General Public License
# along with gprMax. If not, see .
-import sys, os, argparse
+import argparse
+import os
+import sys
+
import h5py
import numpy as np
import matplotlib.pyplot as plt
diff --git a/tests/test_compare_numerical.py b/tests/test_compare_numerical.py
index 338d9073..5d7d5f45 100644
--- a/tests/test_compare_numerical.py
+++ b/tests/test_compare_numerical.py
@@ -16,11 +16,14 @@
# You should have received a copy of the GNU General Public License
# along with gprMax. If not, see .
-import sys, os
+import os
+import sys
+
import h5py
import numpy as np
import matplotlib as mpl
import matplotlib.pyplot as plt
+
from tools.plot_fields import plot_Ascan
"""Compare field outputs
diff --git a/tools/plot_Ascan.py b/tools/plot_Ascan.py
index 9324c839..99815713 100644
--- a/tools/plot_Ascan.py
+++ b/tools/plot_Ascan.py
@@ -16,7 +16,9 @@
# You should have received a copy of the GNU General Public License
# along with gprMax. If not, see .
-import os, argparse
+import argparse
+import os
+
import h5py
import numpy as np
import matplotlib.pyplot as plt
diff --git a/tools/plot_Bscan.py b/tools/plot_Bscan.py
index 5772d227..267d3771 100644
--- a/tools/plot_Bscan.py
+++ b/tools/plot_Bscan.py
@@ -16,7 +16,9 @@
# You should have received a copy of the GNU General Public License
# along with gprMax. If not, see .
-import os, argparse
+import argparse
+import os
+
import h5py
import numpy as np
import matplotlib.pyplot as plt
diff --git a/tools/plot_antenna_params.py b/tools/plot_antenna_params.py
index 3868f80f..6c1c6077 100644
--- a/tools/plot_antenna_params.py
+++ b/tools/plot_antenna_params.py
@@ -16,7 +16,9 @@
# You should have received a copy of the GNU General Public License
# along with gprMax. If not, see .
-import argparse, os
+import argparse
+import os
+
import h5py
import numpy as np
import matplotlib.pyplot as plt
diff --git a/tools/plot_source_wave.py b/tools/plot_source_wave.py
index 301fb199..b269ce8d 100644
--- a/tools/plot_source_wave.py
+++ b/tools/plot_source_wave.py
@@ -16,7 +16,9 @@
# You should have received a copy of the GNU General Public License
# along with gprMax. If not, see .
-import os, argparse
+import argparse
+import os
+
import numpy as np
import matplotlib.pyplot as plt