change the import slightly to try and fix the tests for python 3.3