fixed minor ui bug with key