diff --git a/plot.py b/plot.py
index 8ff24d8..38358e7 100644
--- a/plot.py
+++ b/plot.py
@@ -6,8 +6,8 @@ metrics = pandas.read_csv('./metrics.csv',
                               encoding='ISO-8859-1',
                               sep=',')
 
-builds = metrics[0]
-accuracy = metrics[1]
+builds = metrics[metrics.columns[0]]
+accuracy = metrics[metrics.columns[1]]
 
 
 plt.plot(builds,accuracy)