fix the bug

fix the bug