improving MainCtrl Unit test