diff --git a/asterix-transactions/src/main/java/edu/uci/ics/asterix/transaction/management/opcallbacks/IndexOperationTracker.java b/asterix-transactions/src/main/java/edu/uci/ics/asterix/transaction/management/opcallbacks/IndexOperationTracker.java
index 6713492..a34e6dd 100644
--- a/asterix-transactions/src/main/java/edu/uci/ics/asterix/transaction/management/opcallbacks/IndexOperationTracker.java
+++ b/asterix-transactions/src/main/java/edu/uci/ics/asterix/transaction/management/opcallbacks/IndexOperationTracker.java
@@ -91,8 +91,7 @@
     }
 
     private AbstractOperationCallback getOperationCallback(ILSMIndexOperationContext opCtx) {
-        IndexOperation op = opCtx.getOperation();
-        if (op == IndexOperation.SEARCH || op == IndexOperation.DISKORDERSCAN) {
+        if (opCtx.getSearchOperationCallback() != null) {
             return (AbstractOperationCallback) opCtx.getSearchOperationCallback();
         } else {
             return (AbstractOperationCallback) opCtx.getModificationCallback();