Make sure python2 is still compatible