Consider the group $U(3,p^2)$ of $3\times 3$ upper-triangular matrices whose diagonal entries are $1$ and entries are from field of order $p^2$. Then subgroup $$\begin{Bmatrix} \begin{bmatrix} 1 & 0 & *\\ & 1 & 0 \\ & & 1\end{bmatrix}\end{Bmatrix} $$ with $*$ varying over the field, is equal to the center of the group, so it is a normal subgroup of order $p^2$.
Question: Is it the only normal subgroup of order $p^2$?
In general, if we replace field of order $p^2$ by finite field of higher order $p^k$, does similar conclusion holds (that normal subgroup of order equal to order of the center is center itself?)