Add nan check