Fixed missing modules in setup.py