diff --git a/zipline/algorithm.py b/zipline/algorithm.py
index 1c17bde9..4eb709e4 100644
--- a/zipline/algorithm.py
+++ b/zipline/algorithm.py
@@ -100,7 +100,7 @@ class TradingAlgorithm(object):
         context.sid = 'AAPL'
         context.amount = 100
 
-    def handle_data(self, data):
+    def handle_data(context, data):
         sid = context.sid
         amount = context.amount
         order(sid, amount)