Include missed main.h change.