minor fix at base.py