Fixup missed import