fixup: add include that got lost in rebasing