Why did you remove the nan part?