diff --git a/ec2driver.py b/ec2driver.py
index 4fccc87..a183d78 100644
--- a/ec2driver.py
+++ b/ec2driver.py
@@ -292,7 +292,7 @@ class EC2Driver(driver.ComputeDriver):
                                    block_device_info=None):
         pass
 
-    def finish_revert_migration(self, instance, network_info,
+    def finish_revert_migration(self, context, instance, network_info,
                                 block_device_info=None, power_on=True):
         pass