diff --git a/Project.toml b/Project.toml
index 0653a8e..0502c80 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,7 +1,7 @@
 name = "OpenSpiel"
 uuid = "ceb70bd2-fe3f-44f0-b81f-41608acaf2f2"
 authors = ["Jun Tian <tianjun.cpp@gmail.com>"]
-version = "0.2.1"
+version = "0.2.2"
 
 [deps]
 CxxWrap = "1f15a43c-97ca-5a2a-ae31-89f07a497df4"
diff --git a/src/patch.jl b/src/patch.jl
index ca4f7d0..d5a0c4b 100644
--- a/src/patch.jl
+++ b/src/patch.jl
@@ -90,6 +90,8 @@ information_state_string(state::CxxWrap.StdLib.UniquePtrAllocated{State}) = info
 
 information_state_string(state::CxxWrap.StdLib.UniquePtrAllocated{State}, i) = information_state_string(state[], i)
 
+information_state_tensor(state::CxxWrap.StdLib.UniquePtrAllocated{State}, i::Int64) = information_state_tensor(state[], i)
+
 get_uniform_policy(game::CxxWrap.StdLib.SharedPtrAllocated{Game}) = get_uniform_policy(game[])
 
 record_batched_trajectories(game::CxxWrap.StdLib.SharedPtrAllocated{Game}, p::CxxWrap.StdLib.StdVectorAllocated{TabularPolicy}, m::StdMapAllocated{StdString, Int32}, i::Int64, b::Bool, i2::Int64, i3::Int64) = record_batched_trajectories(game[], p, m, i, b, i2, i3)