diff --git a/tests/testthat/test-solve_POMDP.R b/tests/testthat/test-solve_POMDP.R
index fe75caa..6275bce 100644
--- a/tests/testthat/test-solve_POMDP.R
+++ b/tests/testthat/test-solve_POMDP.R
@@ -6,8 +6,7 @@ context("solve_POMDP")
 data("Tiger")
 sol <- solve_POMDP(Tiger)
 expect_identical(nrow(sol$solution$pg[[1]]), 5L)
-#plot_policy_graph(sol, belief = FALSE)
-plot_policy_graph(sol, belief = 100)
+plot_policy_graph(sol)
 
 sol <- solve_POMDP(Tiger, horizon = 3, method = "incprune")
 expect_identical(length(sol$solution$pg), 3L)