diff --git a/.travis.yml b/.travis.yml
index b745923..aa98b10 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -6,10 +6,13 @@ cache:
   npm: true
   directories:
     - node_modules
-before_install:
-  - chmod +x src/postinstall.js
+addons:
+  apt:
+    packages:
+      - graphviz
 install:
   - npm install
+  - npm install node-plantuml
 before_script:
   - npm install -g jshint
 script:
diff --git a/package.json b/package.json
index 9ab4569..63e60c2 100644
--- a/package.json
+++ b/package.json
@@ -18,7 +18,7 @@
   },
   "scripts": {
     "test": "jest",
-    "install": "src/postinstall.js"
+    "install": "node src/postinstall.js"
   },
   "repository": {
     "type": "git",
diff --git a/src/postinstall.js b/src/postinstall.js
index e9754f0..3b68c2b 100644
--- a/src/postinstall.js
+++ b/src/postinstall.js
@@ -14,5 +14,5 @@ ncp('./sample_data/',dir, function(err){
     if(err)
         console.error(err);
     else
-        console.log('done');
+        console.log(`sample data copied to ${dir}`);
 });
\ No newline at end of file