Fix invalid syntax with python 2