diff --git a/build.cake b/build.cake
index a5138151..1de9257f 100644
--- a/build.cake
+++ b/build.cake
@@ -257,6 +257,7 @@ Task("TestCore")
 /// </summary>
 Task("Test")
     .IsDependentOn("Setup")
+	.IsDependentOn("SRGen")
     .IsDependentOn("BuildTest")
     .Does(() =>
 {
@@ -304,6 +305,7 @@ Task("Test")
 /// </summary>
 Task("OnlyPublish")
     .IsDependentOn("Setup")
+	.IsDependentOn("SRGen")
     .Does(() =>
 {
     var project = buildPlan.MainProject;
@@ -534,7 +536,8 @@ Task("SRGen")
 		var dotnetArgs = string.Format("{0} -or \"{1}\" -oc \"{2}\" -ns \"{3}\" -an \"{4}\" -cn SR -l CS -dnx \"{5}\"",
 			srgenPath, outputResx, outputCs, projectName, projectName, projectStrings);
 		Information("{0}", dotnetArgs);
-		Run(dotnetcli, dotnetArgs);
+		Run(dotnetcli, dotnetArgs)
+			.ExceptionOnError("Failed to run SRGen.");
 	}
 });
 
diff --git a/scripts/packages.config b/scripts/packages.config
index 54f9b711..c870b006 100644
--- a/scripts/packages.config
+++ b/scripts/packages.config
@@ -2,5 +2,5 @@
 <packages>
     <package id="Cake" version="0.10.1" />
     <package id="Newtonsoft.Json" version="8.0.3" />
-    <package id="Microsoft.DataTools.SrGen" version="1.0.1-preview" />
+    <package id="Microsoft.DataTools.SrGen" version="1.0.1" />
 </packages>