Fix focus after notification