Update register to use correct object