ModelNotFoundException fix