Fixed bug in get_media test