Add a small fix to let this package work with the Blaze branch.