From 445ed0ed364572d6e6f3cfab509b18b65f77970b Mon Sep 17 00:00:00 2001
From: RomanQed <gbakaldin@gmail.com>
Date: Wed, 21 Feb 2024 21:34:34 +0300
Subject: [PATCH] Improve AbstractProviderBuilder

---
 .../io/github/amayaframework/di/AbstractProviderBuilder.java    | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/main/java/io/github/amayaframework/di/AbstractProviderBuilder.java b/src/main/java/io/github/amayaframework/di/AbstractProviderBuilder.java
index 3ec630b..b763479 100644
--- 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;
     }