Fix CustomRegisterTest.