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 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; }