Fix reference issue.
Merge pull request #476 from gsn/master

Fix reference issue.