small model averaging changes