minor compilation fix (missing import)