Corrected 'class' parameter

Corrected 'class' parameter