flake8 fix
Module renamed