bug fix on gui