adding class to fix e2e test