diff --git a/src/apply.luau b/src/apply.luau
index c05fedf..f8755f5 100644
--- a/src/apply.luau
+++ b/src/apply.luau
@@ -1,4 +1,4 @@
-local typeof = game and typeof or require "../test/mock".typeof :: never
+local typeof = Instance and typeof or require "../test/mock".typeof :: never
 
 local flags = require "./flags"
 local implicit_effect = require "./implicit_effect"
diff --git a/src/cleanup.luau b/src/cleanup.luau
index 6c46bbf..376fdd4 100644
--- a/src/cleanup.luau
+++ b/src/cleanup.luau
@@ -1,4 +1,4 @@
-local typeof = game and typeof or require "../test/mock".typeof :: never
+local typeof = Instance and typeof or require "../test/mock".typeof :: never
 
 local graph = require "./graph"
 local get_scope = graph.get_scope
diff --git a/src/create.luau b/src/create.luau
index 61707b4..94adde1 100644
--- a/src/create.luau
+++ b/src/create.luau
@@ -1,5 +1,5 @@
-local typeof = game and typeof or require "../test/mock".typeof :: never
-local Instance = game and Instance or require "../test/mock".Instance :: never
+local typeof = Instance and typeof or require "../test/mock".typeof :: never
+local Instance = Instance or require "../test/mock".Instance :: never
 
 local defaults = require "./defaults"
 local apply = require "./apply"
diff --git a/src/defaults.luau b/src/defaults.luau
index 77040ff..f268a3f 100644
--- a/src/defaults.luau
+++ b/src/defaults.luau
@@ -1,5 +1,5 @@
-local Enum = game and Enum or require "../test/mock".Enum :: never
-local Color3 = game and Color3 or require "../test/mock".Color3 :: never
+local Enum = Enum or require "../test/mock".Enum :: never
+local Color3 = Color3 or require "../test/mock".Color3 :: never
 
 return {
     Part = {
diff --git a/src/lib.luau b/src/lib.luau
index a8a3885..99fbd7a 100644
--- a/src/lib.luau
+++ b/src/lib.luau
@@ -31,17 +31,17 @@ export type Instances = create.Instances
 export type Properties<T> = create.Properties<T>
 
 local function step(dt: number)
-    if game then debug.profilebegin("VIDE STEP") end
+    if debug.profilebegin then debug.profilebegin("VIDE STEP") end
 
-    if game then debug.profilebegin("VIDE SPRING") end
+    if debug.profilebegin then debug.profilebegin("VIDE SPRING") end
     update_springs(dt)
-    if game then debug.profileend() end
+    if debug.profileend then debug.profileend() end
 
-    if game then debug.profilebegin("VIDE SCHEDULER") end
+    if debug.profilebegin then debug.profilebegin("VIDE SCHEDULER") end
     update_timeouts(dt)
-    if game then debug.profileend() end
+    if debug.profileend then debug.profileend() end
 
-    if game then debug.profileend() end
+    if debug.profileend then debug.profileend() end
 end
 
 local stepped = game and game:GetService("RunService").Heartbeat:Connect(function(dt: number)