FIX: add missing modules to setup.py

We should think about a way to test this ...

Thanks @yuval-harpaz for reporting!