Make tests compatible with unittest discover

Currently running `python -m unittest discover tests` doesn't do anything. Here is a trivial change to fix it.