Refocus input element on click