diff --git a/lib/message.js b/lib/message.js
index 4827e0a..074111d 100644
--- a/lib/message.js
+++ b/lib/message.js
@@ -33,7 +33,7 @@ var events = {
   'onTestEnd': '##teamcity[testFinished flowId=\'[flowId]\' name=\'[name]\']',
   'onTestFail': '##teamcity[testFailed flowId=\'[flowId]\' name=\'[name]\' message=\'[message]\' details=\'[details]\']',
   'onTestSkip': '##teamcity[testIgnored flowId=\'[flowId]\' name=\'[name]\' message=\'pending\']',
-  'onSuiteEnd': '##teamcity[testSuiteFinished flowId=\'[flowId]\' name=\'[name]\']'
+  'onSuiteEnd': '##teamcity[testSuiteFinished flowId=\'[flowId]\' name=\'[name]\']',
 };
 
 exports.buildFormatter = buildFormatter;
diff --git a/package.json b/package.json
index e2c4263..478f2ae 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "wdio-teamcity-reporter",
-  "version": "1.1.1",
+  "version": "2.0.0",
   "description": "WebdriverIO Teamcity reporter",
   "main": "index.js",
   "engines": {
@@ -8,12 +8,8 @@
   },
   "scripts": {
     "cleanup": "git reset --hard",
-    "compile": "babel lib --out-dir lib && babel index.js --out-file index.js",
     "lint": "eslint lib/*.js test/**/*.js *.js",
-    "prepublish": "in-publish && npm run -s compile || in-install",
-    "publish": "in-publish && npm run -s cleanup || in-install",
-    "test": "tape --require babel-register test/unit/*.js",
-    "unit": "tape test/unit/*.js",
+    "test": "tape test/unit/*.js",
     "wdio": "wdio test/integration/wdio.conf.js"
   },
   "repository": {
@@ -37,14 +33,9 @@
     "@wdio/local-runner": "^5.7.0",
     "@wdio/mocha-framework": "^5.7.0",
     "@wdio/reporter": "^5.7.0",
-    "@wdio/selenium-standalone-service": "^5.7.0",
+    "@wdio/selenium-standalone-service": "^5.16.10",
     "@wdio/sync": "^5.7.0",
-    "babel-cli": "^6.11.4",
-    "babel-preset-env": "^1.7.0",
-    "babel-preset-es2015": "^6.13.2",
-    "babel-register": "^6.11.6",
     "eslint": "^6.7.2",
-    "in-publish": "^2.0.0",
     "tape": "^4.6.0"
   },
   "peerDependencies": {
diff --git a/test/integration/dom.js b/test/integration/dom.js
index 1d920f7..c19cac2 100644
--- a/test/integration/dom.js
+++ b/test/integration/dom.js
@@ -8,6 +8,6 @@ suite('simple assertions for the DOM', () => {
   });
 
   test('does body exists', () => {
-    assert.equal(browser.getAttribute('body', 'class'), 'hp vasq');
+    assert.equal(browser.$('body').getAttribute('class'), 'hp vasq');
   });
 });
diff --git a/test/integration/fail.js b/test/integration/fail.js
index 976f7be..9b6abf3 100644
--- a/test/integration/fail.js
+++ b/test/integration/fail.js
@@ -10,6 +10,6 @@ suite('simple exception', () => {
   test.skip('pending test', () => {});
 
   test('artificial error', () => {
-    assert.equal(browser.getAttribute('body', 'non-existing-attribute'), 'guess');
+    assert.equal(browser.$('body').getAttribute('non-existing-attribute'), null);
   });
 });
diff --git a/test/integration/wdio.conf.js b/test/integration/wdio.conf.js
index ede4b45..3cd4bf6 100644
--- a/test/integration/wdio.conf.js
+++ b/test/integration/wdio.conf.js
@@ -7,7 +7,7 @@ exports.config = {
 
   capabilities: [{
     maxInstances: 2,
-    browserName: 'firefox',
+    browserName: 'chrome',
   }],
 
   sync: true,
diff --git a/test/unit/events.js b/test/unit/events.js
index ca7f6dd..c540928 100644
--- a/test/unit/events.js
+++ b/test/unit/events.js
@@ -44,6 +44,6 @@ test('formatter for events', t => {
 });
 
 function assertMsg(util, expectation, result) {
-  util.ok(result.indexOf(expectation) === 0,
+  util.ok(result.startsWith(expectation),
     `message should start from \`${expectation}\` instead got \`${result}\``);
 }