Improve task runner import