Minor UI Bug Fixed