Small bug fix.
New UI design.