Fix bug affecting undiluted model.