This will probably also want to set the `dtype` to `np.int32`