Fix test compatibility.