Made sure we pass the model when calling createMediaUI