GUI update - bug fix