Fixed error from last commit