Small modification to make python import right module depending on version.