change main for test