diff --git a/worker/src/test/java/org/apache/celeborn/service/deploy/worker/memory/MemoryManagerSuiteJ.java b/worker/src/test/java/org/apache/celeborn/service/deploy/worker/memory/MemoryManagerSuiteJ.java
index 39d934b655b..1cb6513a6a0 100644
--- a/worker/src/test/java/org/apache/celeborn/service/deploy/worker/memory/MemoryManagerSuiteJ.java
+++ b/worker/src/test/java/org/apache/celeborn/service/deploy/worker/memory/MemoryManagerSuiteJ.java
@@ -35,11 +35,11 @@ public void testInitMemoryManagerWithInvalidConfig() {
       Assert.fail("MemoryManager initialize should throw exception with invalid configuration");
     } catch (IllegalArgumentException iae) {
       Assert.assertEquals(
-          iae.getMessage(),
           String.format(
-              "Invalid config, " + "{} should be greater than {}",
+              "Invalid config, {} should be greater than {}",
               CelebornConf.WORKER_DIRECT_MEMORY_RATIO_PAUSE_REPLICATE().key(),
-              CelebornConf.WORKER_DIRECT_MEMORY_RATIO_PAUSE_RECEIVE().key()));
+              CelebornConf.WORKER_DIRECT_MEMORY_RATIO_PAUSE_RECEIVE().key()),
+          iae.getMessage());
     } catch (Exception e) {
       Assert.fail("With unexpected exception" + e);
     }