[FIX] flake 8