update to correct class name