Revert test code