Add @Before in tests
Add @Before in tests