Another bug fix in custom.py.