Fix custom node example