diff --git a/src/input-number-controller.js b/src/input-number-controller.js
index cd5005b..9cd065b 100644
--- a/src/input-number-controller.js
+++ b/src/input-number-controller.js
@@ -14,7 +14,10 @@ export class InputNumberController extends Controller {
   }
 
   get string() {
-    return `${parseFloat(this.stateObj.state)}`;
+    if (typeof this.stateObj.attributes.unit_of_measurement === "undefined") {
+        this.stateObj.attributes.unit_of_measurement = "";
+    }
+    return `${parseFloat(this.stateObj.state)} ${this.stateObj.attributes.unit_of_measurement}`;
   }
 
   get isOff() {