diff --git a/test/signature.test.ts b/test/signature.test.ts
index 47209a0..5cfb1fe 100644
--- a/test/signature.test.ts
+++ b/test/signature.test.ts
@@ -1,14 +1,14 @@
 import {test, expect} from 'bun:test'
 
 import {signature} from '../index'
-import {gen_keypair} from "../src/signature";
+import {derive_keypair, gen_keypair, sign, verify} from "../src/signature";
 
 test('base case', async () => {
     const keypair = await gen_keypair()
     const data = new TextEncoder().encode("Message 123 !")
 
-    const sig = await signature.sign(data, keypair[0])
-    const verification = await signature.verify(data, keypair[1], sig)
+    const sig = await sign(data, keypair.privateKey)
+    const verification = await verify(data, keypair.publicKey, sig)
 
     expect(verification).toBe(true)
 })
@@ -17,7 +17,7 @@ test('inverted keys', async () => {
     const keypair = await signature.gen_keypair()
     const data = new TextEncoder().encode("Message 123 !")
 
-    expect(async () => await signature.sign(keypair[1], data)).toThrow()
+    expect(async () => await sign(data, keypair.publicKey)).toThrow()
 })
 
 test('tampered message', async () => {
@@ -26,8 +26,8 @@ test('tampered message', async () => {
     const data2 = new TextEncoder().encode("Message 321 !")
     expect(data1).not.toEqual(data2)
 
-    const sig = await signature.sign(data1, keypair[0])
-    const verification = await signature.verify(data2, keypair[1], sig)
+    const sig = await sign(data1, keypair.privateKey)
+    const verification = await verify(data2, keypair.publicKey, sig)
 
     expect(verification).toBe(false)
 })
@@ -37,8 +37,8 @@ test('different keypair', async () => {
     const keypair2 = await signature.gen_keypair()
     const data = new TextEncoder().encode("Message 123 !")
 
-    const sig = await signature.sign(data, keypair[0])
-    const verification = await signature.verify(data, keypair2[1], sig)
+    const sig = await sign(data, keypair.privateKey)
+    const verification = await verify(data, keypair2.publicKey, sig)
 
     expect(verification).toBe(false)
 })
@@ -47,9 +47,9 @@ test('tampered signature', async () => {
     const keypair = await signature.gen_keypair()
     const data = new TextEncoder().encode("Message 123 !")
 
-    const sig = await signature.sign(data, keypair[0])
+    const sig = await sign(data, keypair.privateKey)
     sig[0] ^= 1
-    const verification = await signature.verify(data, keypair[1], sig)
+    const verification = await verify(data, keypair.publicKey, sig)
 
     expect(verification).toBe(false)
 })