diff --git a/package.json b/package.json
index cb34a73c6..ffedfd89f 100644
--- a/package.json
+++ b/package.json
@@ -6,6 +6,7 @@
   },
   "main": "./index.js",
   "dependencies": {
+    "ast-types": "git+https://github.com/jlongster/ast-types.git",
     "babylon": "git+https://github.com/jlongster/babylon.git#published",
     "minimist": "^1.2.0",
     "recast": "^0.11.18"
diff --git a/tests_config/run_spec.js b/tests_config/run_spec.js
index 013a811b0..11fe6055f 100644
--- a/tests_config/run_spec.js
+++ b/tests_config/run_spec.js
@@ -33,12 +33,12 @@ function run_spec(dirname) {
 
       if (RUN_AST_TESTS) {
         const source = read(dirname + '/' + filename);
-        const ast = parse(source);
+        const ast = removeEmptyStatements(parse(source));
         let prettyprinted = false;
         let ppast;
         let pperr = null;
         try {
-          ppast = parse(prettyprint(source, path));
+          ppast = removeEmptyStatements(parse(prettyprint(source, path)));
         }
         catch(e) {
           pperr = e.stack;
@@ -47,10 +47,8 @@ function run_spec(dirname) {
         test(path + ' parse', () => {
           expect(pperr).toBe(null);
           expect(ppast).toBeDefined();
-          if(RUN_AST_TESTS === "1") {
-            if(ast.errors.length === 0) {
-              expect(removeEmptyStatements(ast)).toEqual(removeEmptyStatements(ppast));
-            }
+          if(ast.errors.length === 0) {
+            expect(ast).toEqual(ppast);
           }
         });
       }