diff --git a/autocnet/graph/edge.py b/autocnet/graph/edge.py
index 13cd0efcf7e0e485b411d1e6eb0f7554f94c6a7e..7b741416702fc6a4174dc3bd5aea2aa6c6472d75 100644
--- a/autocnet/graph/edge.py
+++ b/autocnet/graph/edge.py
@@ -359,7 +359,7 @@ class Edge(dict, MutableMapping):
             raise AttributeError('This edge does not yet have any matches computed.')
 
         matches, mask = self._clean(clean_keys)
-        domain = self.source.handle.raster_size
+        domain = self.source.geodata.raster_size
 
         # Massage the dataframe into the correct structure
         coords = self.source.get_keypoint_coordinates()
diff --git a/autocnet/vis/graph_view.py b/autocnet/vis/graph_view.py
index a5bb4931fb68f3ac803bb2166da497cf697fa769..dfab120ede9004989bcf3dfd21453b7f04bbdbe0 100644
--- a/autocnet/vis/graph_view.py
+++ b/autocnet/vis/graph_view.py
@@ -1,9 +1,6 @@
-import math
 import numpy as np
 import networkx as nx
 
-from autocnet.examples import get_path
-from autocnet.fileio.io_gdal import GeoDataset
 from matplotlib import pyplot as plt
 import matplotlib