fix missing import on prev commit