diff --git a/stimuli/components/grating.py b/stimuli/components/grating.py
index 79cbb6bb..d9958833 100644
--- a/stimuli/components/grating.py
+++ b/stimuli/components/grating.py
@@ -249,6 +249,7 @@ def square_wave(
     frequency=None,
     n_bars=None,
     bar_width=None,
+    orientation="horizontal",
     intensity_bars=(0.0, 1.0),
     period="ignore",
 ):
@@ -287,6 +288,7 @@ def square_wave(
         frequency=frequency,
         n_bars=n_bars,
         bar_width=bar_width,
+        orientation=orientation,
         period=period,
     )
     mask = stim["mask"]