Improve AbstractProviderBuilder

AmayaFramework · Feb 21, 2024 · 445ed0e · 445ed0e
1 parent a498262
commit 445ed0e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/main/java/io/github/amayaframework/di/AbstractProviderBuilder.java b/src/main/java/io/github/amayaframework/di/AbstractProviderBuilder.java
@@ -71,12 +71,12 @@ public <T> ServiceProviderBuilder addService(Artifact artifact,
         Objects.requireNonNull(artifact);
         Objects.requireNonNull(implementation);
         Objects.requireNonNull(wrapper);
-        strong.remove(artifact);
         // Check if the implementation is a child class of an artifact type
         var parent = artifact.getType();
         if (!parent.isAssignableFrom(implementation)) {
             throw new IllegalArgumentException("The implementation is not a child class of the artifact type");
         }
+        strong.remove(artifact);
         any.put(artifact, Entry.of(implementation, wrapper));
         return this;
     }