Tweak MainTest assert statement