Fix nil node error