diff --git a/tools/screenshot-test/firebase.json b/tools/screenshot-test/firebase.json
new file mode 100644
index 00000000000..6a453cfa579
--- /dev/null
+++ b/tools/screenshot-test/firebase.json
@@ -0,0 +1,11 @@
+{
+  "hosting": {
+    "public": "dist",
+    "rewrites": [
+      {
+        "source": "**",
+        "destination": "/index.html"
+      }
+    ]
+  }
+}
diff --git a/tools/screenshot-test/package.json b/tools/screenshot-test/package.json
index 25242b3ad90..9f5ae91cf24 100644
--- a/tools/screenshot-test/package.json
+++ b/tools/screenshot-test/package.json
@@ -10,13 +10,14 @@
   "private": true,
   "dependencies": {
     "@angular/animations": "^4.0.2",
+    "@angular/cdk": "angular/cdk-builds",
     "@angular/common": "^4.0.0",
     "@angular/compiler": "^4.0.0",
     "@angular/compiler-cli": "^4.0.0",
     "@angular/core": "^4.0.0",
     "@angular/forms": "^4.0.0",
     "@angular/http": "^4.0.0",
-    "@angular/material": "^2.0.0-beta.3",
+    "@angular/material": "angular/material2-builds",
     "@angular/platform-browser": "^4.0.0",
     "@angular/platform-browser-dynamic": "^4.0.0",
     "@angular/router": "^4.0.0",