diff --git a/Project.toml b/Project.toml
index d4d6a2d..0653a8e 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.0"
+version = "0.2.1"
 
 [deps]
 CxxWrap = "1f15a43c-97ca-5a2a-ae31-89f07a497df4"
diff --git a/src/patch.jl b/src/patch.jl
index e864c4a..ca4f7d0 100644
--- a/src/patch.jl
+++ b/src/patch.jl
@@ -88,6 +88,8 @@ is_terminal(state::CxxWrap.StdLib.UniquePtrAllocated{State}) = is_terminal(state
 
 information_state_string(state::CxxWrap.StdLib.UniquePtrAllocated{State}) = information_state_string(state[])
 
+information_state_string(state::CxxWrap.StdLib.UniquePtrAllocated{State}, i) = information_state_string(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)