Fix tentative for the crash while unrolling the network tree