Fixed Pyflakes warning regarding missing import