diff --git a/provider/partition-aws/src/test/java/org/opengroup/osdu/partition/provider/aws/service/PartitionServiceImplTest.java b/provider/partition-aws/src/test/java/org/opengroup/osdu/partition/provider/aws/service/PartitionServiceImplTest.java
index 6b9c468ba8e8e568752b6f701f22d4eb9aaaebdc..9dfe8123402f47fbba9d524ca1a00bd5ee3b2fe3 100644
--- a/provider/partition-aws/src/test/java/org/opengroup/osdu/partition/provider/aws/service/PartitionServiceImplTest.java
+++ b/provider/partition-aws/src/test/java/org/opengroup/osdu/partition/provider/aws/service/PartitionServiceImplTest.java
@@ -15,7 +15,6 @@
 
 package org.opengroup.osdu.partition.provider.aws.service;
 
-
 import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
@@ -29,6 +28,7 @@ import org.opengroup.osdu.partition.provider.aws.util.SSMHelper;
 import org.powermock.core.classloader.annotations.PrepareForTest;
 import org.powermock.modules.junit4.PowerMockRunner;
 
+import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.HashMap;
 import java.util.Map;
@@ -86,7 +86,7 @@ public class PartitionServiceImplTest {
 
         when(ssmHelper.partitionExists(any())).thenReturn(false);
         when(ssmHelper.createOrUpdateSecret(any(), any(), any())).thenReturn(true);
-        when(ssmHelper.getPartitionSecrets(any())).thenReturn(this.partitionInfo.getProperties());
+        when(ssmHelper.getSsmParamsPathsForPartition(any())).thenReturn(new ArrayList<String>(this.partitionInfo.getProperties().keySet()));
 
         PartitionInfo partInfo = partService.createPartition(this.partitionInfo.getProperties().get("id").toString(), this.partitionInfo);
         assertTrue(partInfo.getProperties().size() == 3);