fix test compatibility