minor fix for class node