Fix the error reported by @cxj