updated tests
removed unnecessary trait