fix: fix correct classes on elements;