Fixed error caused by last change