Fixed slight correction bug