Fix to prevent class being removed

stops class being removed from tag when media plugin is used