fixed a bug related to missing import