From 90c1ce5d590ddb65179b91051b813ec5e2e2fb1b Mon Sep 17 00:00:00 2001
From: dragonmacher <48328597+dragonmacher@users.noreply.github.com>
Date: Wed, 18 Nov 2020 09:09:56 -0500
Subject: [PATCH] Fix test failing to recent change

---
 .../java/ghidra/graph/visualization/DefaultGraphDisplay.java  | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/Ghidra/Features/GraphServices/src/main/java/ghidra/graph/visualization/DefaultGraphDisplay.java b/Ghidra/Features/GraphServices/src/main/java/ghidra/graph/visualization/DefaultGraphDisplay.java
index 3fd2e8a5cd..9abd56e378 100644
--- a/Ghidra/Features/GraphServices/src/main/java/ghidra/graph/visualization/DefaultGraphDisplay.java
+++ b/Ghidra/Features/GraphServices/src/main/java/ghidra/graph/visualization/DefaultGraphDisplay.java
@@ -1119,7 +1119,7 @@ public class DefaultGraphDisplay implements GraphDisplay {
 		}
 
 		graphMouse = new JgtPluggableGraphMouse(this);
-		viewer.setGraphMouse(graphMouse);
+		vv.setGraphMouse(graphMouse);
 
 		return vv;
 	}
@@ -1144,7 +1144,7 @@ public class DefaultGraphDisplay implements GraphDisplay {
 			// vertices
 			if (e.getStateChange() == ItemEvent.SELECTED) {
 				Collection<AttributedVertex> selectedVertices = getVertices(e.getItem());
-				notifySelectionChanged(new HashSet<AttributedVertex>(selectedVertices));
+				notifySelectionChanged(new HashSet<>(selectedVertices));
 
 				if (selectedVertices.size() == 1) {
 					// if only one vertex was selected, make it the focused vertex