setup.py fix

Hi

I tried to compile the DistAlgo module but looks like the setup.py has some compilation errors. This change fixes that.