Shorten the test code by using setUp