Fix the bug

Fix the bug of special case.