GUI bug fix