update rivet code