setup: added a missing import