diff --git a/Gemfile b/Gemfile
index a07d3334..5efe9335 100644
--- a/Gemfile
+++ b/Gemfile
@@ -26,7 +26,7 @@ platforms :rbx do
   gem 'rubysl', '~> 2.0'
 end
 
-gem 'prometheus-client', '~> 0.10.0.alpha'
+gem 'prometheus-client', '~> 1.0'
 gem 'prometheus_gcstat', git: 'git@github.com:gocardless/prometheus_gcstat_ruby'
 
 gemspec
diff --git a/lib/generators/que/templates/add_que.rb b/lib/generators/que/templates/add_que.rb
index ef02b6ce..26fe28a7 100644
--- a/lib/generators/que/templates/add_que.rb
+++ b/lib/generators/que/templates/add_que.rb
@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-class AddQue < ActiveRecord::Migration
+class AddQue < ActiveRecord::Migration[<%= ActiveRecord::Migration.current_version %>]
   def self.up
     # The current version as of this migration's creation.
     Que.migrate! version: 4