Update to go along with PR in blaze repo