Fixing upstream test init to follow pep8