Quick fix for python script