diff --git a/src/driver/amdxdna/amdxdna_tdr.c b/src/driver/amdxdna/amdxdna_tdr.c
index f5640b3..716fe19 100644
--- a/src/driver/amdxdna/amdxdna_tdr.c
+++ b/src/driver/amdxdna/amdxdna_tdr.c
@@ -6,8 +6,8 @@
 #include "amdxdna_drm.h"
 #include "amdxdna_tdr.h"
 
-int timeout_in_sec = 2;
-module_param(timeout_in_sec, int, 0644);
+uint timeout_in_sec = 2;
+module_param(timeout_in_sec, uint, 0644);
 MODULE_PARM_DESC(timeout_in_sec, "Seconds to timeout and recovery, default 2; 0 - No TDR");
 
 #define TDR_TIMEOUT_JIFF msecs_to_jiffies(timeout_in_sec * 1000)