Fix NaN check in already developed notebooks