Fix the warning issue