Small fix to plotting in GUI