Fix progress getting lost