Oopps, refactor bug in call to update_beta