Add test helper to make test env consistent