Fixed the diagram backend test to work with the new backend interface.