Fix progress dialog