Remove unnecessary code

Thanks to pyflakes.