[FIX] Fix set workspace to setup.py