fixed annoying UI bug...