Fix: use the new path of models