Fix nan detection in normalisation NP pruning