diff --git a/test/sample/test_sample.py b/test/sample/test_sample.py
index d972f4afa..dc31748e9 100644
--- a/test/sample/test_sample.py
+++ b/test/sample/test_sample.py
@@ -18,7 +18,7 @@
 
 
 def gaussian_llh(x):
-    return float(norm.logpdf(x))
+    return float(norm.logpdf(x).item())
 
 
 def gaussian_problem():