fix focus bug