diff --git a/Library/Homebrew/upgrade.rb b/Library/Homebrew/upgrade.rb
index 48a1a363c9..6acb840881 100644
--- a/Library/Homebrew/upgrade.rb
+++ b/Library/Homebrew/upgrade.rb
@@ -1,4 +1,4 @@
-# typed: false
+# typed: true
 # frozen_string_literal: true
 
 require "reinstall"
@@ -209,7 +209,7 @@ module Homebrew
     ensure
       # restore previous installation state if build failed
       begin
-        linked_kegs.each(&:link) if linked_kegs.present? && !f.latest_version_installed?
+        linked_kegs&.each(&:link) unless formula.latest_version_installed?
       rescue
         nil
       end
diff --git a/Library/Homebrew/upgrade.rbi b/Library/Homebrew/upgrade.rbi
new file mode 100644
index 0000000000..602b1d5a2e
--- /dev/null
+++ b/Library/Homebrew/upgrade.rbi
@@ -0,0 +1,5 @@
+# typed: strict
+
+module Homebrew::Upgrade
+  include Kernel
+end