Use the right import path to be compatible with python3.