diff --git a/test/contract_spec.js b/test/contract_spec.js
index 6e6e235..7766b45 100644
--- a/test/contract_spec.js
+++ b/test/contract_spec.js
@@ -158,7 +158,7 @@ contract("ERC20Bucket", function () {
 
   it("return correct bucket type", async function () {
     let bucketType = await ERC20Bucket.methods.bucketType().call();
-    assert(parseInt(bucketType), 20);
+    assert.equal(parseInt(bucketType), 20);
   });
 
   it("shop buys 100 tokens", async function () {
diff --git a/test/nft_contract_spec.js b/test/nft_contract_spec.js
index eb13ae6..db1a703 100644
--- a/test/nft_contract_spec.js
+++ b/test/nft_contract_spec.js
@@ -159,7 +159,7 @@ contract("NFTBucket", function () {
 
   it("return correct bucket type", async function () {
     let bucketType = await NFTBucket.methods.bucketType().call();
-    assert(parseInt(bucketType), 721);
+    assert.equal(parseInt(bucketType), 721);
   });