Fixed the reference issue