-FIXED- Minor UI bug