Fixed teardown class to have attribute