add reg_source support