Adding the className property to the context