Fix predictive key
minor modification for setup.py