Fixed error introduced in last commit