fixed pruning issue