flak8 fixes