fixed bug

fixed the bug