diff --git a/src/test/java/course_tracker_use_cases/GradeCalculatorTest.java b/src/test/java/course_tracker_use_cases/GradeCalculatorTest.java
index 7116f86..7ed4640 100644
--- a/src/test/java/course_tracker_use_cases/GradeCalculatorTest.java
+++ b/src/test/java/course_tracker_use_cases/GradeCalculatorTest.java
@@ -4,7 +4,7 @@
 import org.junit.jupiter.api.Test;
 import screens.course_features.InMemoryCourse;
 import screens.course_tracker.GradeCalculatorPresenter;
-import use_cases.course_features.course_enrolment_use_case.CourseEnrolmentDsGateway;
+import use_cases.course_features.course_enrolment_use_case.CourseEnrolmentCourseDsGateway;
 import use_cases.course_tracker.grade_calculator_use_case.GradeCalculatorInteractor;
 import use_cases.course_tracker.grade_calculator_use_case.GradeCalculatorRequestModel;
 import use_cases.course_tracker.grade_calculator_use_case.GradeCalculatorResponseModel;
@@ -16,7 +16,7 @@
 
 public class GradeCalculatorTest {
 
-    CourseEnrolmentDsGateway courseAccess;
+    CourseEnrolmentCourseDsGateway courseAccess;
 
     /**
      * Helper function to set up all the static variables
diff --git a/src/test/java/course_tracker_use_cases/ProgressTrackerTest.java b/src/test/java/course_tracker_use_cases/ProgressTrackerTest.java
index eda9a3e..0babc15 100644
--- a/src/test/java/course_tracker_use_cases/ProgressTrackerTest.java
+++ b/src/test/java/course_tracker_use_cases/ProgressTrackerTest.java
@@ -4,7 +4,7 @@
 import org.junit.jupiter.api.Test;
 import screens.course_features.InMemoryCourse;
 import screens.course_tracker.ProgressTrackerPresenter;
-import use_cases.course_features.course_enrolment_use_case.CourseEnrolmentDsGateway;
+import use_cases.course_features.course_enrolment_use_case.CourseEnrolmentCourseDsGateway;
 import use_cases.course_tracker.progress_tracker_use_case.ProgressTrackerInteractor;
 import use_cases.course_tracker.progress_tracker_use_case.ProgressTrackerRequestModel;
 import use_cases.course_tracker.progress_tracker_use_case.ProgressTrackerResponseModel;
@@ -18,7 +18,7 @@
 
 public class ProgressTrackerTest {
 
-    CourseEnrolmentDsGateway courseAccess;
+    CourseEnrolmentCourseDsGateway courseAccess;
 
     /**
      * Helper function to set up all the static variables