Hopefully fix the colour bug in the GUI