fix bug
Fix bug.