Update setup.py

Add missing braces