Python bug fix