diff --git a/custom_components/owlet/const.py b/custom_components/owlet/const.py
index b6789f5..c574fe2 100644
--- a/custom_components/owlet/const.py
+++ b/custom_components/owlet/const.py
@@ -8,4 +8,4 @@ CONF_OWLET_REFRESH = "refresh"
 SUPPORTED_VERSIONS = [3]
 POLLING_INTERVAL = 5
 MANUFACTURER = "Owlet Baby Care"
-SLEEP_STATES = {1, "Awake", 8, "Light Sleep", 15, "Deep Sleep"}
+SLEEP_STATES = {1: "Awake", 8: "Light Sleep", 15: "Deep Sleep"}