diff --git a/spring-core-5/src/test/java/com/baeldung/springbean/naming/SpringBeanNamingTest.java b/spring-core-5/src/test/java/com/baeldung/springbean/naming/SpringBeanNamingTest.java
index 9e38bf7ae2..66503f7c26 100644
--- a/spring-core-5/src/test/java/com/baeldung/springbean/naming/SpringBeanNamingTest.java
+++ b/spring-core-5/src/test/java/com/baeldung/springbean/naming/SpringBeanNamingTest.java
@@ -1,5 +1,7 @@
 package com.baeldung.springbean.naming;
 
+import com.baeldung.springbean.naming.component.Cat;
+import com.baeldung.springbean.naming.component.Dog;
 import com.baeldung.springbean.naming.service.PetShow;
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
@@ -64,7 +66,7 @@ public class SpringBeanNamingTest {
         assertNotNull(context.getBean("cat"));
         assertNotNull(context.getBean("dog"));
 
-        assertThat(petShow.getCat().name()).isEqualTo("Cat");
-        assertThat(petShow.getDog().name()).isEqualTo("Dog");
+        assertThat(petShow.getCat().getClass()).isEqualTo(Cat.class);
+        assertThat(petShow.getDog().getClass()).isEqualTo(Dog.class);
     }
 }