diff --git a/source/java/org/alfresco/repo/avm/AVMServiceTest.java b/source/java/org/alfresco/repo/avm/AVMServiceTest.java
index 9839047c38..66b2f3f562 100644
--- a/source/java/org/alfresco/repo/avm/AVMServiceTest.java
+++ b/source/java/org/alfresco/repo/avm/AVMServiceTest.java
@@ -4908,10 +4908,10 @@ public class AVMServiceTest extends AVMServiceTestBase
             assertEquals(1, fService.createSnapshot("main", null, null).get("main").intValue());
             loader.recursiveLoad("source/java/org/alfresco/repo/action", "main:/");
             times.add(System.currentTimeMillis());
-            assertEquals(2, fService.createSnapshot("main", null, null));
+            assertEquals(2, fService.createSnapshot("main", null, null).get("main").intValue());
             loader.recursiveLoad("source/java/org/alfresco/repo/audit", "main:/");
             times.add(System.currentTimeMillis());
-            assertEquals(3, fService.createSnapshot("main", null, null));
+            assertEquals(3, fService.createSnapshot("main", null, null).get("main").intValue());
             assertEquals(1, fService.getStoreVersions("main", null, new Date(times.get(0))).size());
             assertEquals(3, fService.getStoreVersions("main", new Date(times.get(0)), null).size());
             assertEquals(2, fService.getStoreVersions("main", new Date(times.get(1)),