Fix return MediaType annotations