need to fix gui