Fixed reference issue

#2