lint code
fix args on Python 3