diff --git a/test/integration/Provider.test.ts b/test/integration/Provider.test.ts
index 8a6ceb3c..7c0334b7 100644
--- a/test/integration/Provider.test.ts
+++ b/test/integration/Provider.test.ts
@@ -1,6 +1,6 @@
 import { Ocean } from '../../src/ocean/Ocean'
 import config from './config'
-import { assert } from 'console'
+import { assert } from 'chai'
 
 describe('Provider tests', () => {
   let ocean: Ocean
diff --git a/test/unit/ocean/DID.test.ts b/test/unit/ocean/DID.test.ts
index 9765931c..c3154202 100644
--- a/test/unit/ocean/DID.test.ts
+++ b/test/unit/ocean/DID.test.ts
@@ -1,4 +1,4 @@
-import assert from 'assert'
+import { assert } from 'chai'
 import DID from '../../../src/ocean/DID'
 
 describe('DID', () => {