fix a bug about ImportError