Another bit I forgot in the previous PR