diff --git a/tests/educators.test.ts b/tests/educators.test.ts
index 1496b00..4c0adc7 100644
--- a/tests/educators.test.ts
+++ b/tests/educators.test.ts
@@ -7,7 +7,7 @@ import type { Express } from "express";
 
 import { authorize, getTestDatabaseConnection } from "./utils";
 import { setupApp } from "../src/app";
-import { Class, Educator, Student, StudentsClasses } from "../src/models";
+import { Educator } from "../src/models";
 import { createApp } from "../src/server";
 import { v4 } from "uuid";
 
diff --git a/tests/students.test.ts b/tests/students.test.ts
index f2c182b..193de39 100644
--- a/tests/students.test.ts
+++ b/tests/students.test.ts
@@ -27,11 +27,13 @@ async function setupStudentInClasses() {
     name: v4(),
     educator_id: educator.id,
     code: v4(),
+    expected_size: 1,
   });
   const class2 = await Class.create({
     name: v4(),
     educator_id: educator.id,
     code: v4(),
+    expected_size: 1,
   });
   const student = await Student.create({
     email: v4(),