fix focus issue on elements