correction of little part of code