Fix focus on exit