Fixing NaN tests