I would remove this check and class attribute.