diff --git a/GEODE/Visualisation/DensityProfile.py b/GEODE/Visualisation/DensityProfile.py
index 4e8c4bd162fd988c35c70821575e5d322d3d35d5..36745a95566fd32f8267c118964629591463c5d7 100644
--- a/GEODE/Visualisation/DensityProfile.py
+++ b/GEODE/Visualisation/DensityProfile.py
@@ -42,13 +42,16 @@ def densityProfile(measures, resolution):
 def drawDensityProfile(measures, outputFile, resolution):
     plotDensity(densityProfile(measures, resolution), outputFile)
 
-def drawDensityProfileCLI(arguments):
+def getArgs(arguments):
     cli = argparse.ArgumentParser(
             prog='densityProfile',
             description="Draw a density profile from a set of occurrences")
     cli.add_argument('inputTSV')
     cli.add_argument('outputPNG')
     cli.add_argument('-r', '--resolution', type=int)
-    args = cli.parse_args(arguments)
+    return cli.parse_args(arguments)
+
+def drawDensityProfileCLI(arguments):
+    args = getArgs(arguments)
     measures = [m[1] for m in tabular(args.inputTSV).iterrows()]
     drawDensityProfile(measures, args.outputPNG, args.resolution or 100)