Fixed another reference issue