This looked repetitive, overwrites the 'class' attr added in the enter_ method.