Fix minor dialog bug