diff --git a/test/__snapshots__/index.test.ts.snap b/test/__snapshots__/index.test.ts.snap
index 6822631..997c35c 100644
--- a/test/__snapshots__/index.test.ts.snap
+++ b/test/__snapshots__/index.test.ts.snap
@@ -8,63 +8,3 @@ exports[`work fine 1`] = `
   "truncated": false,
 }
 `;
-
-exports[`work fine: partial 1`] = `
-{
-  "token": " swo",
-}
-`;
-
-exports[`work fine: partial 2`] = `
-{
-  "token": "cha",
-}
-`;
-
-exports[`work fine: partial 3`] = `
-{
-  "token": "door",
-}
-`;
-
-exports[`work fine: partial 4`] = `
-{
-  "token": "ter",
-}
-`;
-
-exports[`work fine: partial 5`] = `
-{
-  "token": " scientific",
-}
-`;
-
-exports[`work fine: partial 6`] = `
-{
-  "token": " Windows",
-}
-`;
-
-exports[`work fine: partial 7`] = `
-{
-  "token": "Ca",
-}
-`;
-
-exports[`work fine: partial 8`] = `
-{
-  "token": " occupied",
-}
-`;
-
-exports[`work fine: partial 9`] = `
-{
-  "token": "rå",
-}
-`;
-
-exports[`work fine: partial 10`] = `
-{
-  "token": " alta",
-}
-`;
diff --git a/test/index.test.ts b/test/index.test.ts
index 33ccc3e..ed5cd4f 100644
--- a/test/index.test.ts
+++ b/test/index.test.ts
@@ -11,7 +11,7 @@ it('work fine', async () => {
     n_predict: 10,
     seed: 0,
   }, (data) => {
-    expect(data).toMatchSnapshot('partial')
+    expect(data).toMatchObject({ token: expect.any(String) })
   })
   expect(result).toMatchSnapshot()
 })