Forgot to revert some test code.