Minor change in MainTest