Update setup.py

Fixing issue on sub-folder detection.