Update test

CS2103AUG2016-W10-C2 · Nov 6, 2016 · a85cf7f · a85cf7f
1 parent 526e3ef
commit a85cf7f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/test/java/seedu/priorityq/logic/LogicManagerTest.java b/src/test/java/seedu/priorityq/logic/LogicManagerTest.java
@@ -626,7 +626,7 @@ public void execute_mark_alreadyMarked() throws Exception {
         assertCommandBehavior("mark 1",
                 String.format(MarkCommand.MESSAGE_SUCCESS, alreadyMarked),
                 expectedTM,
-                new ArrayList<>());
+                expectedTM.getTaskList());
     }
 
     @Test
@@ -777,7 +777,7 @@ public void execute_undoUnmark_successful() throws Exception {
         assertCommandBehavior("undo",
                 String.format(UnmarkCommand.MESSAGE_UNDO_SUCCESS, toBeUnmarked),
                 expectedTM,
-                new ArrayList<>());
+                expectedTM.getTaskList());
     }
 
     @Test