Add missing piece of earlier commit.