UI Bug Fix