Fixed error caused by previous commit