fix Image and Video model