Added workaround for graph display bug. Still no idea what's causing it