mocks: fix flake8 issue