a bit change to TestRegister