[Refactoring] Just a nazy correction :D