diff --git a/Rakefile b/Rakefile
index 2193d7d..88700cf 100644
--- a/Rakefile
+++ b/Rakefile
@@ -16,5 +16,9 @@ def config_params
 end
 
 def site_url
-  ENV["DEPLOY_URL"]
+  if ENV["CONTEXT"] == "production"
+    ENV["URL"]
+  else
+    ENV["DEPLOY_URL"]
+  end
 end