diff --git a/config/deploy.rb b/config/deploy.rb
index 94569ff5a..8e9e4b043 100644
--- a/config/deploy.rb
+++ b/config/deploy.rb
@@ -6,3 +6,4 @@ set :scm, :git
 set :deploy_to, "/var/www/danbooru2"
 set :rbenv_ruby, "2.3.1"
 set :linked_dirs, fetch(:linked_dirs, []).push('log', 'tmp/pids', 'tmp/cache', 'tmp/sockets', 'vendor/bundle')
+set :branch, ENV.fetch("branch", "master")
\ No newline at end of file