Fixed wrapper class name and added it to the sample test config