use correct classnames