Fix refocusing on form elements after validation messages are displayed