Fix running on py2