Minor fix to model checking notebook