Fix silly upgrade GUI bug