diff --git a/keydb_extension.go b/keydb_extension.go
index 3c31e46..c5af580 100644
--- a/keydb_extension.go
+++ b/keydb_extension.go
@@ -20,7 +20,7 @@ func init() {
 type KeyDBHandler struct {
 	Address  string `json:"address"`
 	Password string `json:"password"`
-	DB       string `json:"db"`
+	DB       int    `json:"db"`
 	client   *redis.Client
 }