diff --git a/definitions/staging/stg_agent_status.sqlx b/definitions/staging/stg_agent_status.sqlx
index f74ba2c..ac86d8a 100644
--- a/definitions/staging/stg_agent_status.sqlx
+++ b/definitions/staging/stg_agent_status.sqlx
@@ -1,10 +1,10 @@
 config {
   type: "incremental",  
-  uniqueKey: ["AGENT_ID", "TIMESTAMP"],
+  uniqueKey: ["AGENT_ID", "TIMESTAMP", "STATE"],
   schema: "pphe_five9_stg",  
   assertions: {
-    uniqueKey: ["AGENT_ID", "TIMESTAMP"],
-    nonNull: ["AGENT_ID", "TIMESTAMP"]
+    uniqueKey: ["AGENT_ID", "TIMESTAMP", "STATE"],
+    nonNull: ["AGENT_ID", "TIMESTAMP", "STATE"]
   },
   dependencies: ["agent_status_data_freshness"],
   description: "All history data for agent status"
@@ -14,7 +14,7 @@ config {
 with agent_status_deduplicated as (
 select  *,
         ROW_NUMBER()
-          OVER (PARTITION BY AGENT_ID, TIMESTAMP)
+          OVER (PARTITION BY AGENT_ID, TIMESTAMP, STATE)
           row_number
 from ${ref("pphe_five9_raw","agent_status_*")}
 ${ when(incremental(), `where _TABLE_SUFFIX = FORMAT_DATE('%Y%m%d',current_date("Europe/Amsterdam"))`) }) --last date table is taken