Minor fix to container dialog hiding