small fix of import