flak8 fix