Change the way the media class gets set up