diff --git a/docs/unity/part-2.md b/docs/unity/part-2.md
index 9e9936c9..16ab0950 100644
--- a/docs/unity/part-2.md
+++ b/docs/unity/part-2.md
@@ -215,7 +215,7 @@ public partial struct Circle
     public uint player_id;
     public DbVector2 direction;
     public float speed;
-    public ulong last_split_time;
+    public SpacetimeDB.Timestamp last_split_time;
 }
 
 [Table(Name = "food", Public = true)]
@@ -509,7 +509,7 @@ public class GameManager : MonoBehaviour
 
         // If the user has a SpacetimeDB auth token stored in the Unity PlayerPrefs,
         // we can use it to authenticate the connection.
-		if (PlayerPrefs.HasKey(AuthToken.GetTokenKey()))
+		if (PlayerPrefs.HasKey(AuthToken.Token != ""))
         {
             builder = builder.WithToken(AuthToken.Token);
         }
@@ -548,7 +548,7 @@ public class GameManager : MonoBehaviour
         }
     }
 
-    private void HandleSubscriptionApplied(EventContext ctx)
+    private void HandleSubscriptionApplied(SubscriptionEventContext ctx)
     {
         Debug.Log("Subscription applied!");
         OnSubscriptionApplied?.Invoke();
diff --git a/docs/unity/part-3.md b/docs/unity/part-3.md
index ecd1990a..5e4ce6fe 100644
--- a/docs/unity/part-3.md
+++ b/docs/unity/part-3.md
@@ -491,7 +491,7 @@ public static Entity SpawnPlayerInitialCircle(ReducerContext ctx, uint player_id
     );
 }
 
-public static Entity SpawnCircleAt(ReducerContext ctx, uint player_id, uint mass, DbVector2 position, DateTimeOffset timestamp)
+public static Entity SpawnCircleAt(ReducerContext ctx, uint player_id, uint mass, DbVector2 position, SpacetimeDB.Timestamp timestamp)
 {
     var entity = ctx.Db.entity.Insert(new Entity
     {
@@ -505,7 +505,7 @@ public static Entity SpawnCircleAt(ReducerContext ctx, uint player_id, uint mass
         player_id = player_id,
         direction = new DbVector2(0, 1),
         speed = 0f,
-        last_split_time = (ulong)timestamp.ToUnixTimeMilliseconds(),
+        last_split_time = timestamp,
     });
     return entity;
 }