fix for reference
fixed test
Merge pull request #17 from zigorou/fix_reference

fixed #6