Fixed a minor bug with the last commit