Fix typos.
Use the reshaped array.