Fix issue with adding classes to the body