diff --git a/api/controller/artifacts_unittest.py b/api/controller/artifacts_unittest.py
index 57e72df..9630a6d 100644
--- a/api/controller/artifacts_unittest.py
+++ b/api/controller/artifacts_unittest.py
@@ -692,19 +692,15 @@
         )
 
         actual = [
-            os.path.basename(artifact.artifact_path.path)
+            os.path.relpath(artifact.artifact_path.path, self.archive_root)
             for artifact in self.output_proto.artifacts
         ]
         expected = ["payload1.bin", "payload2.bin"]
         self.assertCountEqual(actual, expected)
 
         actual = [
-            os.path.basename(path)
-            for path in osutils.DirectoryIterator(
-                os.path.dirname(
-                    self.output_proto.artifacts[0].artifact_path.path
-                )
-            )
+            os.path.relpath(path, self.archive_root)
+            for path in osutils.DirectoryIterator(self.archive_root)
         ]
         self.assertCountEqual(actual, expected)