Fix program location association in the trace formula on branching