fix the reference issue