diff --git a/zipline/finance/trading.py b/zipline/finance/trading.py
index 6c3d1318..e6119a8e 100644
--- a/zipline/finance/trading.py
+++ b/zipline/finance/trading.py
@@ -29,6 +29,8 @@ class TradeSimulationClient(qmsg.Component):
         )
         
         self.perf = perf.PerformanceTracker(self.trading_environment)
+        self.on_done = self.perf.handle_simulation_end
+        
     
     @property
     def get_id(self):