fixed issue with last commit