bug fix (replace right class tag)