diff --git a/bottle_mongo.py b/bottle_mongo.py
index 73e2551..8c86781 100644
--- a/bottle_mongo.py
+++ b/bottle_mongo.py
@@ -155,9 +155,9 @@ def setup(self, app):
                     app.uninstall(other)
                     return
 
-    def apply(self, callback, context):
+    def apply(self, callback, route):
         """Return a decorated route callback."""
-        args = inspect.getargspec(context.callback)[0]
+        args = inspect.getargspec(route.callback)[0]
         # Skip this callback if we don't need to do anything
         if self.keyword not in args:
             return callback