fixed small mistake in prod.py