Fixed failing unit test by copying tests.py from author's repo.