Updated migration to actually overwrite media_type