fix: improve heuristic
2 files changed