fix: fix heuristics