Update of the wrapper class

Misplaced comment