fix missing import removed by mistake