fix rigister bug