diff --git a/tests/plots/colormap.py b/tests/plots/colormap.py
index b30b525..ddaedf5 100644
--- a/tests/plots/colormap.py
+++ b/tests/plots/colormap.py
@@ -2,9 +2,13 @@
 
 import numpy as np
 import matplotlib.pyplot as plt
+import sys
 from matplotlib import cm
 
-filename='infnoise.bin'
+if sys.argv[1]:
+	filename=sys.argv[1]
+else:
+	filename='infnoise.bin'
 
 nx = 1000
 ny = 1000
@@ -25,4 +29,4 @@ cbar = plt.colorbar(cax, ticks=[255, 127, 0])
 cbar.ax.set_yticklabels(['255', '127', '0'])
 
 plt.savefig(filename + '-colormap.png')
-plt.show() 
+plt.show()
diff --git a/tests/plots/scatterplot.py b/tests/plots/scatterplot.py
index 5e430a7..a89543c 100644
--- a/tests/plots/scatterplot.py
+++ b/tests/plots/scatterplot.py
@@ -2,8 +2,13 @@
 
 import matplotlib.pyplot as plt
 import numpy as np
+import sys
+
+if sys.argv[1]:
+        filename=sys.argv[1]
+else:
+        filename='infnoise.bin'
 
-filename='infnoise.bin'
 
 bp = np.dtype([('byte1',np.uint8),('byte2',np.uint8)]) # 'struct' byte pairs