Fix flake8 in task helper

Change-Id: I9fb023b10f7fea1a8d8d2572cde16aa19433a495