Fix py2 compat (ggggggrrrrrrrrr I hate py2)