Fixing breaking syntax on python 3.X