Correcting deprecated model