  We improve the $L^p(\mathbb{R}^n)$ bounds on Stein's square function to the
best-known range of the Fourier restriction problem when $n\geq4$. Applications
including certain local smoothing estimates are also discussed.
