Fix WYS focus problem