Fixes pyflakes' list comprehension error

More details:

https://groups.google.com/forum/#!topic/pylearn-dev/jDqu2PAJuzk