Change focus after editing