GUI bugs fix