diff --git a/tests/test_wave.py b/tests/test_wave.py
index ab0c1c6..7b3e539 100644
--- a/tests/test_wave.py
+++ b/tests/test_wave.py
@@ -9,7 +9,7 @@
 @frames_comparison()
 def test_linearwave(scene):
     wave = LinearWave()
-    wave.set_time(2)
+    wave.set(time=2)
     scene.add(wave)
 
 
@@ -21,7 +21,7 @@ def test_radialwave(scene):
         checkerboard_colors=[BLUE_D],
         stroke_width=0,
     )
-    wave.set_time(2)
+    wave.set(time=2)
     scene.add(wave)
 
 
@@ -37,3 +37,4 @@ def test_standingwave(scene):
     for wave in waves:
         wave.start_wave()
     scene.wait()
+