Fix error caused by the last commit