fix the small bug

Hi,Dr Ji

    I have fixed the bug.