Forgot to include the focus back.