Remove test code from task runner