diff --git a/features/lib/table_diff.js b/features/lib/table_diff.js
index 4acbd23c1..76389bac4 100644
--- a/features/lib/table_diff.js
+++ b/features/lib/table_diff.js
@@ -50,5 +50,5 @@ module.exports = function (expected, actual) {
         s.push(rowString);
     });
 
-    return s.join('\n') + '\nTODO this is a temp workaround waiting for https://github.com/cucumber/cucumber-js/issues/534';
+    return s.join('\n');
 };
diff --git a/features/support/data.js b/features/support/data.js
index b5cc95924..c01cda3b7 100644
--- a/features/support/data.js
+++ b/features/support/data.js
@@ -248,7 +248,7 @@ module.exports = function () {
         q.awaitAll((err, actual) => {
             if (err) return callback(err);
             let diff = tableDiff(table, actual);
-            if (diff) callback(new Error(diff));
+            if (diff) callback(diff);
             else callback();
         });
     };