Using correct class parameter.