Dialog bug fix