fix reference issue