Bug fix in correction script.