Fix the missed issue