Fix broken build

Missed a rename in setup.py that broke the build.