diff --git a/test/unit/collection-proxy.test.ts b/test/unit/collection-proxy.test.ts
index f3e9e51..c7d428c 100644
--- a/test/unit/collection-proxy.test.ts
+++ b/test/unit/collection-proxy.test.ts
@@ -1,12 +1,13 @@
 import { expect } from "../test-helper"
 import { Person } from "../fixtures"
 import { JsonapiCollectionDoc, JsonapiResource } from "../../src/index"
+import { PersistedSpraypaintRecord } from "../../src/index"
 
 import { CollectionProxy } from "../../src/proxies/collection-proxy"
 
 describe("CollectionProxy", () => {
   let personData: JsonapiCollectionDoc
-  let recordArray: Person[]
+  let recordArray: PersistedSpraypaintRecord<Person>[]
 
   beforeEach(() => {
     personData = {
diff --git a/test/unit/record-proxy.test.ts b/test/unit/record-proxy.test.ts
index f99675c..27c23f0 100644
--- a/test/unit/record-proxy.test.ts
+++ b/test/unit/record-proxy.test.ts
@@ -1,10 +1,11 @@
 import { expect } from "../test-helper"
 import { Person } from "../fixtures"
 import { RecordProxy } from "../../src/proxies"
+import { PersistedSpraypaintRecord } from "../../src/index"
 import { JsonapiResourceDoc, JsonapiResource } from "../../src/index"
 
 describe("RecordProxy", () => {
-  let modelRecord: Person
+  let modelRecord: PersistedSpraypaintRecord<Person>
   let personData: JsonapiResourceDoc
 
   beforeEach(() => {