diff --git a/src/rpi-cpu2mqtt.py b/src/rpi-cpu2mqtt.py
index 5bc98ef..c5b4bec 100644
--- a/src/rpi-cpu2mqtt.py
+++ b/src/rpi-cpu2mqtt.py
@@ -127,7 +127,7 @@ def check_rpi5_fan_speed():
 def get_os():
     full_cmd = 'cat /etc/os-release | grep -i pretty_name'
     pretty_name = subprocess.Popen(full_cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE).communicate()[0].decode("utf-8")
-    pretty_name = pretty_name.split('=')[1].replace('"', '')
+    pretty_name = pretty_name.split('=')[1].replace('"', '').replace('\n', '')
 
     return(pretty_name)
 
@@ -289,7 +289,8 @@ def config_json(what_config):
         data["state_class"] = "measurement"
     elif what_config == "git_update":
         data["icon"] = "mdi:git"
-        data["name"] = "Update Available"
+        data["name"] = "RPi MQTT Monitor"
+        data["device_class"] = "update"
         data["state_class"] = "measurement"
         data["payload_on"] = "on"
         data["payload_off"] = "off"