Refactoring test - made setUp()