Pyflakes fix for build