Remove className reference