diff --git a/fabfile.py b/fabfile.py
index 070fd4cda..b3144d8ac 100644
--- a/fabfile.py
+++ b/fabfile.py
@@ -56,17 +56,15 @@ def test():
             local('py.test -x')
 
 
-def train(train_loc=None, dev_loc=None, model_dir=None):
-    if train_loc is None:
-        train_loc = 'corpora/en/ym.wsj02-21.conll'
-    if dev_loc is None:
-        dev_loc = 'corpora/en/ym.wsj24.conll'
+def train(json_dir=None, dev_loc=None, model_dir=None):
+    if json_dir is None:
+        json_dir = 'corpora/en/json'
     if model_dir is None:
         model_dir = 'models/en/'
     with virtualenv(VENV_DIR):
         with lcd(path.dirname(__file__)):
             local('python bin/init_model.py lang_data/en/ corpora/en/ ' + model_dir)
-            local('python bin/parser/train.py %s %s %s' % (train_loc, dev_loc, model_dir))
+            local('python bin/parser/train.py %s %s' % (json_dir, model_dir))
 
 
 def travis():