UI styles bug fixed