fixed tests that failed when run using 'python -m unittest discover'