Fix references
Merge pull request #12 from jrogelstad/master

Fix references