Fixed a reference issue