Fixed a set comprehension that doesn't work in python 2.6