Fix Python 3 imports

Some Python 3-style import I missed in my previous PR  >.<