Fix __module__ attribute of node classes.