Please use `np.newaxis` instead of just None.