diff --git a/astyanax-entity-mapper/src/main/java/com/netflix/astyanax/entitystore/DefaultEntityManager.java b/astyanax-entity-mapper/src/main/java/com/netflix/astyanax/entitystore/DefaultEntityManager.java index d51d4cad0..70fe21391 100644 --- a/astyanax-entity-mapper/src/main/java/com/netflix/astyanax/entitystore/DefaultEntityManager.java +++ b/astyanax-entity-mapper/src/main/java/com/netflix/astyanax/entitystore/DefaultEntityManager.java @@ -233,7 +233,7 @@ private DefaultEntityManager(Builder builder) { public void put(T entity) throws PersistenceException { try { lifecycleHandler.onPrePersist(entity); - MutationBatch mb = newMutationBatch(); + MutationBatch mb = getMutationBatch(); entityMapper.fillMutationBatch(mb, columnFamily, entity); if (autoCommit) mb.execute(); @@ -284,7 +284,7 @@ public void remove(T entity) throws PersistenceException { try { lifecycleHandler.onPreRemove(entity); id = entityMapper.getEntityId(entity); - MutationBatch mb = newMutationBatch(); + MutationBatch mb = getMutationBatch(); mb.withRow(columnFamily, id).delete(); if (autoCommit) mb.execute();