unit: add missing imports