Fixing the NaN bug