Change class name to user the wrapper