Fixed error introduced by previous commit