Fixed error introduced by the last commit