Fix revert tests