buff.py fix minor issue