Fixed a error from last commit