Remove hack for media gen.