Update base.py

The function takes one argument.