BUG: fix python error in script