Fix NA bug