Update vag.py

fix