Fix custom register
Merge pull request #546 from liuchuangww/user

Fix custom register