diff --git a/package.json b/package.json
index 40edfe0..acbec54 100644
--- a/package.json
+++ b/package.json
@@ -41,8 +41,8 @@
 		"@types/jest": "^21.1.6",
 		"@types/lodash": "^4.14.85",
 		"@types/node": "^8.0.53",
-		"@types/react": "^16.0.25",
 		"@types/puppeteer": "^0.13.0",
+		"@types/react": "^16.0.25",
 		"awesome-typescript-loader": "^3.4.0",
 		"css-loader": "^0.28.7",
 		"dagre": "^0.7.4",
@@ -51,6 +51,7 @@
 		"node-sass": "^4.7.2",
 		"prettier": "^1.8.2",
 		"puppeteer": "^0.13.0",
+		"raf": "^3.4.0",
 		"react-dom": "^16.1.1",
 		"react-test-renderer": "^16.1.1",
 		"sass-loader": "^6.0.6",
diff --git a/tests/2.test.tsx b/tests/2.test.tsx
index 85cac03..b27a2a6 100644
--- a/tests/2.test.tsx
+++ b/tests/2.test.tsx
@@ -1,3 +1,5 @@
 import initStoryshots from '@storybook/addon-storyshots';
 
-initStoryshots({ /* configuration options */ });
\ No newline at end of file
+import 'raf/polyfill';
+
+initStoryshots({ /* configuration options */ });