Fix flake8 error - relative import