minor 2-line change to main brought up while doing code review