From 827a1ea44a003393172230931728d7e7de0a14cb Mon Sep 17 00:00:00 2001
From: Alec Larson <1925840+aleclarson@users.noreply.github.com>
Date: Tue, 6 Dec 2022 22:24:46 -0500
Subject: [PATCH] fix: use `addRefreshWrapper` even if only $RefreshSig$ is
 used

---
 packages/plugin-react/src/index.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/packages/plugin-react/src/index.ts b/packages/plugin-react/src/index.ts
index 314ac3402..294716ba6 100644
--- a/packages/plugin-react/src/index.ts
+++ b/packages/plugin-react/src/index.ts
@@ -335,7 +335,7 @@ export default function viteReact(opts: Options = {}): PluginOption[] {
 
         if (result) {
           let code = result.code!
-          if (useFastRefresh && /\$RefreshReg\$\(/.test(code)) {
+          if (useFastRefresh && /\$Refresh(Sig|Reg)\$\(/.test(code)) {
             code = addRefreshWrapper(code, id)
           }
           return {