Running `./setup.py test` works again

Fix 72a1b9bafb4503918d31491e5a22341190252d38.