Fix AB pruning