Fix reference issue