diff --git a/cmd/db/main.go b/cmd/db/main.go
index 205d8bf..765cad0 100644
--- a/cmd/db/main.go
+++ b/cmd/db/main.go
@@ -60,9 +60,7 @@ func AddInDB(dbName string, line string) error {
 func DeleteInDB(dbName, id string) error {
 	key := []byte(dbName + "/" + id + "/bash")
 
-	return DB.Update(func(txn *badger.Txn) error {
-		return txn.Delete(key)
-	})
+	return kv.DeleteInDB(DB, string(key))
 }
 
 func CountInDB(dbName, id string) int {