Fix focus after close dialog