Same as above, should probably use the `TYPES` from the model.