fixed the bug

fixed the bug