diff --git a/test/cases/dbUpgrade.ts b/test/cases/dbUpgrade.ts
index 8bcc744..c8de4a6 100644
--- a/test/cases/dbUpgrade.ts
+++ b/test/cases/dbUpgrade.ts
@@ -1,10 +1,14 @@
+import assert from "assert";
+
 import {db, privateDB} from "../../src/databases/databases";
 
 describe("dbUpgrade", () => {
     it("Should update the database version when starting the application", async () => {
         const dbVersion = (await db.prepare("get", "SELECT key, value FROM config where key = ?", ["version"])).value;
         const privateVersion = (await privateDB.prepare("get", "SELECT key, value FROM config where key = ?", ["version"])).value;
-        if (dbVersion >= 1 && privateVersion >= 1) return;
-        else return `Versions are not at least 1. db is ${dbVersion}, private is ${privateVersion}`;
+
+        assert(
+            dbVersion >= 1 && privateVersion >= 1,
+            `Versions are not at least 1. db is ${dbVersion}, private is ${privateVersion}`);
     });
 });
diff --git a/test/cases/getLockCategoriesByHash.ts b/test/cases/getLockCategoriesByHash.ts
index 710611e..d6ce0b2 100644
--- a/test/cases/getLockCategoriesByHash.ts
+++ b/test/cases/getLockCategoriesByHash.ts
@@ -24,8 +24,9 @@ describe("getLockCategoriesByHash", () => {
 
     it("Database should be greater or equal to version 20", async () => {
         const version = (await db.prepare("get", "SELECT key, value FROM config where key = ?", ["version"])).value;
-        if (version >= 20) return;
-        else return `Version isn't greater than 20. Version is ${version}`;
+        assert(
+            version >= 20,
+            `Version isn't greater than 20. Version is ${version}`);
     });
 
     it("Should be able to get multiple locks in one object", (done: Done) => {