diff --git a/GEODE/Visualisation/ConfusionMatrix.py b/GEODE/Visualisation/ConfusionMatrix.py
index b3c099c70336f0a38a2e16b9195148fb6fda6f9b..47cec22426a7f584b48e11813881e33448e403e5 100644
--- a/GEODE/Visualisation/ConfusionMatrix.py
+++ b/GEODE/Visualisation/ConfusionMatrix.py
@@ -38,7 +38,7 @@ def confusionMatrix(data, labels):
     matrix = confusion_matrix(truth, answers, labels=labels, normalize='true')
     return {'matrix': matrix.tolist(), 'labels': labels}
 
-def getConfusionMatrix(inputFile, labels):
+def getConfusionMatrix(inputFile, labels=None):
     if inputFile[-4:] == '.tsv':
         return confusionMatrix(tabular(inputFile), labels)
     elif inputFile[-5:] == '.json':