[Gaurav] fixing flake8 issue