correction bug

correction bug