Fixed the error from previous commit