diff --git a/app/src/test/java/lightcontainer/storage/RepositoryTests.java b/app/src/test/java/lightcontainer/storage/RepositoryTests.java
index d94a0f7..543ee54 100644
--- a/app/src/test/java/lightcontainer/storage/RepositoryTests.java
+++ b/app/src/test/java/lightcontainer/storage/RepositoryTests.java
@@ -35,14 +35,13 @@ public class RepositoryTests {
         appConfig.setMulticastPort(15502);
         appConfig.setNetworkInterface("My network interface");
         appConfig.setTls(false);
-        Map<String, File> files = new HashMap<>();
         Set<String> storage = new HashSet<>();
         storage.add("StorBackEnd1");
         File file1 = new File("File1", 15, "8d8d8d8d", storage);
-        files.put(file1.getName(), file1);
         appData.setAppConfig(appConfig);
         appData.addUser("User1", "Password", "djdjjdj");
         JsonAdapter jsonAdapter = new JsonAdapter(appData);
+        appData.addFileFor(file1, "User1");
         //WHEN Repository calls save method
         String filePath = "src/test/resources/test.json";
         Repository repository = new Repository(filePath, jsonAdapter);