diff --git a/tests/test_sine.py b/tests/test_sine.py
index 5513c8e2..dc2523f8 100644
--- a/tests/test_sine.py
+++ b/tests/test_sine.py
@@ -60,7 +60,7 @@ def test_sine():
             plot_to_tensorboard(writer, "Plot/train", i)
 
         # Test observation
-        obs = dataset._generate_observation(size)
+        obs = dataset.generate_observation(size)
         plt.cla()
         plot_evaluation(model, dataset, obs)
         plot_observation(obs)