fix another missing import