Revert test changes