Fixing bug that the previous commit made.