Fix require for TaskMain in test