diff --git a/test/vomit.js b/test/vomit.js
index ec45a13..507ef15 100644
--- a/test/vomit.js
+++ b/test/vomit.js
@@ -19,3 +19,12 @@ tape('should create multiple elements', (test) => {
   var section = vomit`<section><button>${label}</button></section>`
   test.equal(section.outerHTML, '<section><button>hello</button></section>')
 })
+
+
+tape('should wrap function', (test) => {
+  test.plan(1)
+  var btn = vomit(function(data) {
+    return vomit`<button>${data}</button>`
+  })
+  test.equal(btn('hello').outerHTML, '<button>hello</button>')
+})
\ No newline at end of file