Add py2 compat bullshit