patch assertion.py for latest CustomFunction