diff --git a/test/tc_mark_as_done.vim b/test/tc_mark_as_done.vim
index 6328a9c..23fc4ce 100644
--- a/test/tc_mark_as_done.vim
+++ b/test/tc_mark_as_done.vim
@@ -1,6 +1,7 @@
 let s:here = expand('<sfile>:p:h')
-let s:tc = unittest#testcase#new('Mask As Done', 
-      \ { 'data': s:here . '/tc_mark_as_done.todo.txt' })
+let s:context = todo#txt#__context__()
+let s:context['data'] = s:here . '/tc_mark_as_done.todo.txt'
+let s:tc = unittest#testcase#new('Mark As Done', s:context)
 
 let s:TODAY = strftime("%Y-%m-%d")
 
@@ -18,7 +19,7 @@ let s:ALL_TASKS_DONE = [
 
 function! s:tc.test_mark_as_done()
   call self.data.goto('lorem_ipsum')
-  call self.data.execute('call todo#txt#mark_as_done()')
+  call todo#txt#mark_as_done()
   call self.assert_equal(s:FIRST_TASK_DONE, self.data.get('lorem_ipsum'))
 endfunction