diff --git a/client-spark/common/src/main/java/org/apache/spark/shuffle/reader/RssShuffleDataIterator.java b/client-spark/common/src/main/java/org/apache/spark/shuffle/reader/RssShuffleDataIterator.java
index 1cee994315..e1b97896f0 100644
--- a/client-spark/common/src/main/java/org/apache/spark/shuffle/reader/RssShuffleDataIterator.java
+++ b/client-spark/common/src/main/java/org/apache/spark/shuffle/reader/RssShuffleDataIterator.java
@@ -151,15 +151,7 @@ public boolean hasNext() {
   }
 
   private boolean isSameMemoryType(ByteBuffer left, ByteBuffer right) {
-    if (left.isDirect() && right.isDirect()) {
-      return true;
-    }
-
-    if (!left.isDirect() && !right.isDirect()) {
-      return true;
-    }
-
-    return false;
+    return left.isDirect() == right.isDirect();
   }
 
   private int uncompress(CompressedShuffleBlock rawBlock, ByteBuffer rawData) {