Made pyflakes changes to appease Jenkins

Pyflakes didn't like my change.