Fix GUI bugs