Update testing case to make DI nodes have the correct format.