fix node class saving