fixing compilation issue with missing import