fixed test.py: using same heuristic despite label