Small gui bug fix