Revert new test class also