diff --git a/lib/validates_timeliness/mongoid/info.rb b/lib/validates_timeliness/mongoid/info.rb
index 1d9b5df..9a88b7f 100644
--- a/lib/validates_timeliness/mongoid/info.rb
+++ b/lib/validates_timeliness/mongoid/info.rb
@@ -2,7 +2,7 @@
 
 module ValidatesTimeliness
   module Mongoid
-    VERSION = '5.0.0'
+    VERSION = '5.0.1'
 
     # Name to gem
     GEM_NAME = 'validates_timeliness-mongoid'
diff --git a/validates_timeliness-mongoid.gemspec b/validates_timeliness-mongoid.gemspec
index 2e4dc5a..c00044d 100644
--- a/validates_timeliness-mongoid.gemspec
+++ b/validates_timeliness-mongoid.gemspec
@@ -6,10 +6,12 @@ $LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
 require 'validates_timeliness/mongoid/info'
 
 Gem::Specification.new do |spec|
-  spec.version       = if ENV['GITHUB_REF'].eql?('refs/heads/develop')
-                         "#{ValidatesTimeliness::Mongoid::VERSION}.pre.#{ENV['GITHUB_RUN_ID']}"
+  version = ValidatesTimeliness::Mongoid::VERSION
+  spec.version       = if ENV.key?('GITHUB_RUN_ID') &&
+                          !ENV['GITHUB_REF'].eql?('refs/heads/master')
+                         "#{version}.pre.#{ENV['GITHUB_RUN_ID']}"
                        else
-                         ValidatesTimeliness::Mongoid::VERSION
+                         version
                        end
 
   spec.name          = ValidatesTimeliness::Mongoid::GEM_NAME