fix a missing import