Fix the complie error

caused by inconsistent template parameters.