Fixing minor bug in GUI