Fix break on Python 2