Set focus after update.