Fix GUI bug