Fix gen_media test