Minor fix for the correct model inherutance