add missing import and missing var