Fixing test case: Oversaw reference to tx when refactoring to use graph throughout.