Fixed the same bug with error message