fix Python UDF