From 55b74e81aa2bfab05fac8dab9859a4853eb9f49d Mon Sep 17 00:00:00 2001
From: David <davidaase@hotmail.com>
Date: Mon, 1 May 2017 16:54:05 +0200
Subject: [PATCH] Improve generics in Attr.addTo

---
 src/main/java/j2html/attributes/Attr.java | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/src/main/java/j2html/attributes/Attr.java b/src/main/java/j2html/attributes/Attr.java
index d56f659..36712d3 100644
--- a/src/main/java/j2html/attributes/Attr.java
+++ b/src/main/java/j2html/attributes/Attr.java
@@ -33,18 +33,17 @@ public class Attr {
         return new Shortform(id.trim(), classes.toString().trim());
     }
 
-    @SuppressWarnings("unchecked")
-    public static <T> T addTo(Tag<? extends Tag> tag, Attr.Shortform shortform) {
+    public static <T extends Tag<T>> T addTo(T tag, Attr.Shortform shortform) {
         if (!"".equals(shortform.id) && !"".equals(shortform.classes)) {
-            return (T) tag.withId(shortform.id).withClass(shortform.classes);
+            return tag.withId(shortform.id).withClass(shortform.classes);
         }
         if (!"".equals(shortform.id)) {
-            return (T) tag.withId(shortform.id);
+            return tag.withId(shortform.id);
         }
         if (!"".equals(shortform.classes)) {
-            return (T) tag.withClass(shortform.classes);
+            return tag.withClass(shortform.classes);
         }
-        return (T) tag;
+        return tag;
     }
 
     private Attr() {