minor changes, fixing flake