Fix register() usage to pass tests.