Updated return types for Media fields