Change class rendering.

Instead of relying on an attribute that is not always there, check for the passed widget and decide which class to add.