Fix flake8

Change-Id: Ie5a69e403a8bac0c27d796cbf97875e3df10ba19