This fix is to be able to run the test as a python script (no py.test)