killbill-memoizeit

diff --git a/entitlement/src/main/java/com/ning/billing/entitlement/engine/dao/ISubscriptionSqlDao.java b/entitlement/src/main/java/com/ning/billing/entitlement/engine/dao/ISubscriptionSqlDao.java
index 83ce0fe..e1af556 100644
--- a/entitlement/src/main/java/com/ning/billing/entitlement/engine/dao/ISubscriptionSqlDao.java
+++ b/entitlement/src/main/java/com/ning/billing/entitlement/engine/dao/ISubscriptionSqlDao.java
@@ -57,7 +57,7 @@ public interface ISubscriptionSqlDao extends Transactional<ISubscriptionSqlDao>,
     public List<ISubscription> getSubscriptionsFromBundleId(@Bind("bundle_id") String bundleId);
 
     @SqlUpdate
-    public void updateSubscription(@Bind("active_version") long activeVersion, @Bind("ctd_dt") Date ctd, @Bind("ptd_dt") Date ptd);
+    public void updateSubscription(@Bind("id") String id, @Bind("active_version") long activeVersion, @Bind("ctd_dt") Date ctd, @Bind("ptd_dt") Date ptd);
 
     public static class ISubscriptionDaoBinder implements Binder<Bind, Subscription> {
 

diff --git a/entitlement/src/main/resources/com/ning/billing/entitlement/engine/dao/ISubscriptionSqlDao.sql.stg b/entitlement/src/main/resources/com/ning/billing/entitlement/engine/dao/ISubscriptionSqlDao.sql.stg
index 1be11b8..6d4394d 100644
--- a/entitlement/src/main/resources/com/ning/billing/entitlement/engine/dao/ISubscriptionSqlDao.sql.stg
+++ b/entitlement/src/main/resources/com/ning/billing/entitlement/engine/dao/ISubscriptionSqlDao.sql.stg
@@ -52,11 +52,12 @@ getSubscriptionsFromBundleId(bundle_id) ::= <<
     ;
 >>
 
-updateSubscription(active_version, ctd_dt, ptd_dt) ::= <<
+updateSubscription(id, active_version, ctd_dt, ptd_dt) ::= <<
     update subscriptions
     set
       active_version = :active_version
       , ctd_dt = :ctd_dt
       , ptd_dt = :ptd_dt
+    where id = :id
     ;
 >>