Fix revert testing