fix typo in regression code