Modify register unit test.