diff --git a/src/example/java/io/github/sashirestela/cleverclient/example/StreamExample.java b/src/example/java/io/github/sashirestela/cleverclient/example/StreamExample.java
index 3e24aba..b9770ce 100644
--- a/src/example/java/io/github/sashirestela/cleverclient/example/StreamExample.java
+++ b/src/example/java/io/github/sashirestela/cleverclient/example/StreamExample.java
@@ -12,7 +12,8 @@
  * Before running this example you must have an OpenAI account and keep your Api Key in an
  * environment variable called OPENAI_API_KEY.
  * 
- * @see <a href="https://platform.openai.com/docs/api-reference/authentication">OpenAI Authentication</a>
+ * @see <a href="https://platform.openai.com/docs/api-reference/authentication">OpenAI
+ *      Authentication</a>
  */
 public class StreamExample {
 
diff --git a/src/main/java/io/github/sashirestela/cleverclient/support/ReturnType.java b/src/main/java/io/github/sashirestela/cleverclient/support/ReturnType.java
index 53c0272..b2b8d15 100644
--- a/src/main/java/io/github/sashirestela/cleverclient/support/ReturnType.java
+++ b/src/main/java/io/github/sashirestela/cleverclient/support/ReturnType.java
@@ -48,12 +48,12 @@ public ReturnType(Method method) {
     }
 
     private Map<String, Class<?>> calculateClassByEvent(StreamType[] streamTypeList) {
-        Map<String, Class<?>> classByEvent = new ConcurrentHashMap<>();
+        Map<String, Class<?>> map = new ConcurrentHashMap<>();
         Arrays.stream(streamTypeList).forEach(streamType -> {
             Arrays.stream(streamType.events())
-                    .forEach(event -> classByEvent.put(CleverClientSSE.EVENT_HEADER + event, streamType.type()));
+                    .forEach(event -> map.put(CleverClientSSE.EVENT_HEADER + event, streamType.type()));
         });
-        return classByEvent;
+        return map;
     }
 
     public String toString() {
diff --git a/src/test/java/io/github/sashirestela/cleverclient/support/ReturnTypeTest.java b/src/test/java/io/github/sashirestela/cleverclient/support/ReturnTypeTest.java
index 395b31e..bae5b3d 100644
--- a/src/test/java/io/github/sashirestela/cleverclient/support/ReturnTypeTest.java
+++ b/src/test/java/io/github/sashirestela/cleverclient/support/ReturnTypeTest.java
@@ -13,7 +13,6 @@
 
 import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertNull;
-import static org.junit.jupiter.api.Assertions.assertTrue;
 
 class ReturnTypeTest {
 
@@ -72,7 +71,7 @@ void shouldReturnMapClassByEventWhenTheMethodIsAnnotatedWithStreamType()
         expectedMap.put(CleverClientSSE.EVENT_HEADER + "first.create", First.class);
         expectedMap.put(CleverClientSSE.EVENT_HEADER + "first.complete", First.class);
         expectedMap.put(CleverClientSSE.EVENT_HEADER + "second.create", Second.class);
-        assertTrue(expectedMap.equals(actualMap));
+        assertEquals(Boolean.TRUE, expectedMap.equals(actualMap));
     }
 
     static interface TestInterface {