Fixed error from last change