Fixed GUI problem