Small bug in UI fixed