instead just set the classes directly on the custom element

As far as I can tell, this fixes #1346 (or it was fixed already)