fix shitty heuristics bug