fixed reference issue with last commit