added traits to tests