fixed an issue with the last commit