#46 Fix bug: handle focus after restarting game