Minor tweak for Main