diff --git a/yarp/templates/template.rb b/yarp/templates/template.rb
index 59a779be1f..944cb7edc5 100755
--- a/yarp/templates/template.rb
+++ b/yarp/templates/template.rb
@@ -4,7 +4,7 @@ require "erb"
 require "fileutils"
 require "yaml"
 
-if File.exist?(File.expand_path("../lib/yarp/version", __dir__))
+if File.exist?(File.expand_path("../lib/yarp/version.rb", __dir__))
   # Within the gem/local repository
   require_relative "../lib/yarp/version"
 else