fixing the nan errors