Update jai.py

typo fixed