Update setup.py to inline with master

This should hopefully mean that build will pass with enum dependency met