diff --git a/files/test.vhost b/templates/test.vhost.erb
similarity index 85%
rename from files/test.vhost
rename to templates/test.vhost.erb
index ec9c811..ff3152d 100644
--- a/files/test.vhost
+++ b/templates/test.vhost.erb
@@ -4,8 +4,8 @@
 NameVirtualHost *:80
 <VirtualHost *:80>
   ServerName testvhost
-  DocumentRoot /tmp/testvhost
-  <Directory /tmp/testvhost> 
+  DocumentRoot <%= docroot %>
+  <Directory <%= docroot %>> 
     Options Indexes FollowSymLinks MultiViews
     AllowOverride None
     Order allow,deny
diff --git a/tests/vhost.pp b/tests/vhost.pp
index c916596..6cda916 100644
--- a/tests/vhost.pp
+++ b/tests/vhost.pp
@@ -1,2 +1,6 @@
 include apache
-apache::vhost { 'test.vhost': source => 'puppet:///modules/apache/test.vhost' }
+apache::vhost { 'test.vhost':
+  port     => 80,
+  docroot  => '/tmp/testvhost',
+  template => 'apache/test.vhost.erb'
+}