Using unittest.setup did not work well