Makes some import statement compatible with both python2 and 3.