diff --git a/js/components/main.vue b/js/components/main.vue
index 843450e..46557bd 100644
--- a/js/components/main.vue
+++ b/js/components/main.vue
@@ -162,14 +162,15 @@ var mainWindow = {
                     'time': t,
                     'name': course.name,
                     'temp': false,
-                    'number': course.number
+                    'number': course.number,
+                    'class': course.class
                 })
             }
         },
         'removeCourse': function (course) {
             console.log("remove " + course.name)
             for (var i = this.selectCourses.length - 1; i >= 0; i--) {
-                if (this.selectCourses[i].number == course.number) {
+                if (this.selectCourses[i].number === course.number && this.selectCourses[i].class === course.class) {
                     this.selectCourses.splice(i, 1)
                 }
             }
diff --git a/js/components/table.vue b/js/components/table.vue
index 68f1495..475e848 100644
--- a/js/components/table.vue
+++ b/js/components/table.vue
@@ -53,6 +53,7 @@ var courseTable = {
                 this.courses[c.time] = {
                     'name': c.name,
                     'number': c.number,
+                    'class': c.class,
                     'temp': c.temp
                 }