fixed Zsolts bug in prod.py