diff --git a/Project.toml b/Project.toml
index 0426dfeb..9c8ea4ba 100644
--- a/Project.toml
+++ b/Project.toml
@@ -23,7 +23,7 @@ ArviZSampleChainsDynamicHMCExt = ["SampleChains", "SampleChainsDynamicHMC"]
 ArviZSampleChainsExt = "SampleChains"
 
 [compat]
-InferenceObjects = "0.3.11"
+InferenceObjects = "0.3.11, 0.4"
 MCMCChains = "6"
 MCMCDiagnosticTools = "0.3.4"
 PSIS = "0.9.1"
diff --git a/docs/Project.toml b/docs/Project.toml
index 501fe69a..bd519cc6 100644
--- a/docs/Project.toml
+++ b/docs/Project.toml
@@ -37,7 +37,7 @@ Distributions = "0.25"
 Documenter = "1"
 Downloads = "1.6"
 EvoTrees = "0.14.8, 0.15, 0.16"
-InferenceObjects = "0.3.11"
+InferenceObjects = "0.3.11, 0.4"
 MCMCDiagnosticTools = "0.3"
 MLJBase = "1"
 MLJIteration = "0.6"