fix correction bug