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

Minor fix