Fix missing class for component markup