Fix flake8 'OK import' err