Minor graphical fix