fixing compilation error about missing import