Polish GUI update - Fix issue pointed out by LordBlick