minor reference issue fix