Fix Python 2/3 mismatch