Fixing issue with reference

Fixes #4