diff --git a/package-lock.json b/package-lock.json
index d2b6ffc..b7fed6c 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -5613,11 +5613,13 @@
         },
         "balanced-match": {
           "version": "1.0.0",
-          "bundled": true
+          "bundled": true,
+          "optional": true
         },
         "brace-expansion": {
           "version": "1.1.11",
           "bundled": true,
+          "optional": true,
           "requires": {
             "balanced-match": "^1.0.0",
             "concat-map": "0.0.1"
@@ -5630,15 +5632,18 @@
         },
         "code-point-at": {
           "version": "1.1.0",
-          "bundled": true
+          "bundled": true,
+          "optional": true
         },
         "concat-map": {
           "version": "0.0.1",
-          "bundled": true
+          "bundled": true,
+          "optional": true
         },
         "console-control-strings": {
           "version": "1.1.0",
-          "bundled": true
+          "bundled": true,
+          "optional": true
         },
         "core-util-is": {
           "version": "1.0.2",
@@ -5741,7 +5746,8 @@
         },
         "inherits": {
           "version": "2.0.3",
-          "bundled": true
+          "bundled": true,
+          "optional": true
         },
         "ini": {
           "version": "1.3.5",
@@ -5751,6 +5757,7 @@
         "is-fullwidth-code-point": {
           "version": "1.0.0",
           "bundled": true,
+          "optional": true,
           "requires": {
             "number-is-nan": "^1.0.0"
           }
@@ -5763,17 +5770,20 @@
         "minimatch": {
           "version": "3.0.4",
           "bundled": true,
+          "optional": true,
           "requires": {
             "brace-expansion": "^1.1.7"
           }
         },
         "minimist": {
           "version": "0.0.8",
-          "bundled": true
+          "bundled": true,
+          "optional": true
         },
         "minipass": {
           "version": "2.2.4",
           "bundled": true,
+          "optional": true,
           "requires": {
             "safe-buffer": "^5.1.1",
             "yallist": "^3.0.0"
@@ -5790,6 +5800,7 @@
         "mkdirp": {
           "version": "0.5.1",
           "bundled": true,
+          "optional": true,
           "requires": {
             "minimist": "0.0.8"
           }
@@ -5862,7 +5873,8 @@
         },
         "number-is-nan": {
           "version": "1.0.1",
-          "bundled": true
+          "bundled": true,
+          "optional": true
         },
         "object-assign": {
           "version": "4.1.1",
@@ -5872,6 +5884,7 @@
         "once": {
           "version": "1.4.0",
           "bundled": true,
+          "optional": true,
           "requires": {
             "wrappy": "1"
           }
@@ -5977,6 +5990,7 @@
         "string-width": {
           "version": "1.0.2",
           "bundled": true,
+          "optional": true,
           "requires": {
             "code-point-at": "^1.0.0",
             "is-fullwidth-code-point": "^1.0.0",
diff --git a/package.json b/package.json
index 02e8a10..8570ff6 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "storybook-typescript-react-library",
-  "version": "0.1.0",
+  "version": "0.2.1",
   "main": "build/lib/index.js",
   "types": "build/lib/index.d.ts",
   "files": [
diff --git a/src/ExampleButton/__tests__/__snapshots__/button.test.tsx.snap b/src/ExampleButton/__tests__/__snapshots__/button.test.tsx.snap
index e8a3c18..e913252 100644
--- a/src/ExampleButton/__tests__/__snapshots__/button.test.tsx.snap
+++ b/src/ExampleButton/__tests__/__snapshots__/button.test.tsx.snap
@@ -1,11 +1,7 @@
 // Jest Snapshot v1, https://goo.gl/fbAQLP
 
 exports[`Link changes the class when hovered 1`] = `
-<button
-  className="btn btn-default"
-  disabled={false}
-  type="button"
->
+<button>
   Cute button!
 </button>
 `;
diff --git a/tsconfig.json b/tsconfig.json
index 8ab2c23..7d867f9 100644
--- a/tsconfig.json
+++ b/tsconfig.json
@@ -66,6 +66,7 @@
   "exclude": [
     "build",
     "node_modules",
-    "src/stories"
+    "src/**/*.stories.tsx",
+    "src/**/*.test.tsx"
   ]
 }
\ No newline at end of file