Fixing a reference error
Merge pull request #230 from avoidwork/edge

Minor fix