Fixing 2 reference errors
Merge pull request #49 from avoidwork/edge

Fixing 2 reference errors