diff --git a/ec2/spark_ec2.py b/ec2/spark_ec2.py
index 4d49c6070309637c963327c59817eb37cfb7a29a..677c8f5452296475204eb20cad18a88b73652254 100755
--- a/ec2/spark_ec2.py
+++ b/ec2/spark_ec2.py
@@ -408,9 +408,7 @@ def setup_cluster(conn, master_nodes, slave_nodes, opts, deploy_ssh_key):
 
   # NOTE: We should clone the repository before running deploy_files to
   # prevent ec2-variables.sh from being overwritten
-  # TODO: Before being merged this should be replaced with the correct repo,
-  #       and likely a new branch (to allow backwards compatibility).
-  ssh(master, opts, "rm -rf spark-ec2 && git clone https://github.com/pwendell/spark-ec2.git -b ec2-updates")
+  ssh(master, opts, "rm -rf spark-ec2 && git clone https://github.com/mesos/spark-ec2.git -b v2")
 
   print "Deploying files to master..."
   deploy_files(conn, "deploy.generic", opts, master_nodes, slave_nodes, modules)