diff --git a/src/ha.h b/src/ha.h
index 5886ee6..ee64123 100644
--- a/src/ha.h
+++ b/src/ha.h
@@ -325,7 +325,7 @@ namespace Ha {
         static unordered_map<string, Command*> mapRestoreStateTopics;
 
         StatefulCommand(const char* name, const char* id, const char* type, onMessage f)
-            : Command(name, id, "number", f), StateConfig(this) {}
+            : Command(name, id, type, f), StateConfig(this) {}
 
         void restoreFromState() {
             mapRestoreStateTopics.insert({stateTopic, this});