diff --git a/keydb_extension.go b/keydb_extension.go
index 4b31605..63463f2 100644
--- a/keydb_extension.go
+++ b/keydb_extension.go
@@ -76,15 +76,9 @@ func (h *KeyDBHandler) UnmarshalCaddyfile(d *caddyfile.Dispenser) error {
 					return d.ArgErr()
 				}
 			case "db":
-				var dbStr string
-				if !d.Args(&dbStr) {
+				if !d.Args(strconv.Itoa(h.DB)) {
 					return d.ArgErr()
 				}
-				db, err := strconv.Atoi(dbStr)
-				if err != nil {
-					return d.Errf("invalid db number: %v", err)
-				}
-				h.DB = db
 			}
 		}
 	}