Small GUI fix