diff --git a/scripts/demo_oscillator_simple.py b/scripts/demo_oscillator_simple.py
index 8f2fdbe..326434e 100644
--- a/scripts/demo_oscillator_simple.py
+++ b/scripts/demo_oscillator_simple.py
@@ -118,16 +118,19 @@ def demo_oscillator(waveform: str = "sine", frequency: float = 1.0, frames: int
 if __name__ == "__main__":
     parser = argparse.ArgumentParser(description="Oscillator sensor demo")
     parser.add_argument(
-        "--frames",
-        type=int,
-        default=0,
-        help="Number of frames to render (0 = infinite until Ctrl+C)",
+        "--waveform",
+        choices=["sine", "square", "sawtooth", "triangle", "noise"],
+        default="sine",
+        help="Waveform type",
     )
     parser.add_argument(
         "--frequency", type=float, default=1.0, help="Oscillator frequency in Hz"
     )
     parser.add_argument(
-        "--frames", type=int, default=100, help="Number of frames to render"
+        "--frames",
+        type=int,
+        default=0,
+        help="Number of frames to render (0 = infinite until Ctrl+C)",
     )
 
     args = parser.parse_args()