Fix bad Task import in connection.py