Specific changes that are needed to do this part of https://github.com/quantumlib/Qualtran/issues/873