Fixed reference issue