Fix retry dialog bug