fixed issue in last commit