Fix np.concatenate calls