Fix allocation mismatches for birch.