Fixed python issue