Fixed error from previous commit