Fix focus after closing modal