Fix use new register api