Fixed a little error form the last commit