some fixes into trainBurstEdge