Quantikz gategroup, stepsize isn't functioning properly

Question

I would like the dashed line to extend to the right of the last Hadamard gate on the top register. Everything is, more or less, happening in the top 3 registers

\begin{tikzpicture}
\node[scale=0.88]{
\begin{quantikz}        
        & & & & &  & & & &\lstick{$\ket{0}$}&\qw         &\gate{H}\gategroup[wires = 3,steps = 1, style={dashed, rounded corners, inner sep=4pt}]{\textsc{swap} test}              &\ctrl{2}    &\gate{H} &\qw     &\meter{}\\
        & & & & &  & & & &\lstick{$\ket{0}$}&\qwbundle{t}&\gate{\mathcal{D}[W]} &\targX{}    &\qw&\qw        \rstick{$\frac{1}{\Delta}\sum_j\ket{\omega_j}$} \\
        \lstick{$\ket{0}$}     &\qwbundle{t}        &\gate[2, label style={rotate=90}]{\textsc{qpe}} &\qw\push{\ket{\lambda_0}}       & \qw          &\qw   &\qw                   &\qw &\qw&\qw&\gate[4, label style={rotate=90}]{\textsc{sub}}&\qw &\targX{}    &\qw&\qw        \rstick{$\sum_k d_{0,k}\ket{\omega_{0,k}}$}\\
        \lstick{$\ket{\psi_0}$}&\qwbundle{n}        &              &\qw\push{\ket{\psi_0}}    &\gate[2]{U_D} &\qw     &\push{\sum_k d_{0,k}\ket{\psi_k}} &\gate[3, label style={rotate=90}]{\textsc{qpe}}&\qw\push{\sum_k d_{0,k}\ket{\psi_k}}\\
        \lstick{$\ket{0}$}     &\qwbundle{k}        &\qw           &\qw                                      &\qw     &\push{\ket{0}^{\otimes k}}\hspace{-.33cm}\vcw{-1} \\
        \lstick{$\ket{0}$}     &\qwbundle{t}        &\qw           &\qw                        &\qw          &\qw     &\qw                  &\qw                                     &\push{\sum_k d_{0,k}\ket{\lambda_k}}         &\qw &\qw &\push{\sum_k d_{0,k}\ket{\lambda_{k}}}   
\end{quantikz}  
};
\end{tikzpicture}

Generates:

However, when I change the step size to any number > 1, I get this:

I've been trying to get this to go for a few hours now and I feel like there must be something silly happening here I can't get sorted.

Thanks