From 2a7f8441377ef968bb256d9e4bad50a5067f1284 Mon Sep 17 00:00:00 2001
From: ClaytonTDM <claytontdm@gmail.com>
Date: Fri, 27 Dec 2024 01:17:01 -0500
Subject: [PATCH] fix

Signed-off-by: ClaytonTDM <claytontdm@gmail.com>
---
 game/iframe.html | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/game/iframe.html b/game/iframe.html
index f174d66..84aa154 100644
--- a/game/iframe.html
+++ b/game/iframe.html
@@ -82,9 +82,16 @@ <h2>⚠️ Different Website Warning ⚠️</h2>
 		</div>
 		<script>
 			(function checkSite() {
+				// First check referrer - if it exists and is official site, we're good
+				if (document.referrer && document.referrer.includes('roblnet13.github.io')) {
+					return; // Exit early if referrer is from official site
+				}
+				
+				// Otherwise check current URL only if referrer is empty/missing
 				var currentUrl = window.location.href;
-				var isOfficialSite = currentUrl.includes('roblnet13.github.io') || document.referrer.includes('roblnet13.github.io');
+				var isOfficialSite = !document.referrer && currentUrl.includes('roblnet13.github.io');
 				
+				// Show warning if neither check passed
 				if (!isOfficialSite) {
 					document.getElementById('stolen-warning').style.display = 'flex';
 				}