diff --git a/package.json b/package.json
index 6e82b7c..e2c136a 100644
--- a/package.json
+++ b/package.json
@@ -13,7 +13,7 @@
     "node": ">=0.10.0"
   },
   "scripts": {
-    "test": "node test.js"
+    "test": "ava"
   },
   "files": [
     "index.js"
@@ -37,6 +37,6 @@
     "humanize-url": "^1.0.0"
   },
   "devDependencies": {
-    "ava": "0.0.4"
+    "ava": "*"
   }
 }
diff --git a/test.js b/test.js
index 498e11e..7d71aec 100644
--- a/test.js
+++ b/test.js
@@ -1,14 +1,12 @@
-'use strict';
-var test = require('ava');
-var filenamifyUrl = require('./');
+import test from 'ava';
+import m from './';
 
-test(function (t) {
-	t.assert(filenamifyUrl('sindresorhus.com') === 'sindresorhus.com');
-	t.assert(filenamifyUrl('sindresorhus.com/foo/bar?foo=bar&baz=faz') === 'sindresorhus.com!foo!bar!baz=faz&foo=bar');
-	t.assert(filenamifyUrl('sindresorhus.com/foo/bar?foo=bar&baz=faz') === 'sindresorhus.com!foo!bar!baz=faz&foo=bar');
-	t.assert(filenamifyUrl('http://user:pass@www.sindresorhus.com/foo/bar/') === 'sindresorhus.com!foo!bar');
-	t.assert(filenamifyUrl('http://user@sindresorhus.com') === 'sindresorhus.com');
-	t.assert(filenamifyUrl('sindresorhus.com/foo', {replacement: '🐴'}) === 'sindresorhus.com🐴foo');
-	t.assert(filenamifyUrl('http://www.sindresorhus.com/?query=pageres*|<>:"\\') === 'sindresorhus.com!query=pageres');
-	t.end();
+test(t => {
+	t.is(m('sindresorhus.com'), 'sindresorhus.com');
+	t.is(m('sindresorhus.com/foo/bar?foo=bar&baz=faz'), 'sindresorhus.com!foo!bar!baz=faz&foo=bar');
+	t.is(m('sindresorhus.com/foo/bar?foo=bar&baz=faz'), 'sindresorhus.com!foo!bar!baz=faz&foo=bar');
+	t.is(m('http://user:pass@www.sindresorhus.com/foo/bar/'), 'sindresorhus.com!foo!bar');
+	t.is(m('http://user@sindresorhus.com'), 'sindresorhus.com');
+	t.is(m('sindresorhus.com/foo', {replacement: '🐴'}), 'sindresorhus.com🐴foo');
+	t.is(m('http://www.sindresorhus.com/?query=pageres*|<>:"\\'), 'sindresorhus.com!query=pageres');
 });