1 files changed, 4 insertions, 9 deletions
diff --git a/engines/lab/special.cpp b/engines/lab/special.cpp
index 7d2212138b..74625a63f4 100644
--- a/engines/lab/special.cpp
+++ b/engines/lab/special.cpp
@@ -47,10 +47,6 @@
 
 namespace Lab {
 
-#ifdef GAME_TRIAL
-int g_IsRegistered;
-#endif
-
 extern bool nopalchange, DoBlack, IsHiRes;
 
 extern BitMap *DispBitMap, *DrawBitMap;
@@ -450,12 +446,11 @@ static void changeTile(uint16 col, uint16 row) {
 		return;
 #endif
 
-#if defined(GAME_TRIAL)
-
-		if (!g_IsRegistered)
+		if (g_lab->getFeatures() & GF_WINDOWS_TRIAL) {
+			GUI::MessageDialog trialMessage("This puzzle is not available in the trial version of the game");
+			trialMessage.runModal();
 			return;
-
-#endif
+		}
 
 		check = true;
 		row   = 0;