Added working setUp and tearDown for test.