Fix unit test for main so that we're not testing actual call